R her öğe için sütun ve satır belirten alt öğe elemanları

Question

Her bir öğenin hem satırını hem de sütununu belirten bir matrisi nasıl ayırmalıyım? Şu anda sapply kullanıyorum ama bunun özellikle zarif bulmuyorum:

> mat <- data.frame(a=c(1,2,3),b=c(7,6,5)) 
> mat 
    a b 
1 1 7 
2 2 6 
3 3 5 
> rowSel <- 1:3 
> colSel <- c(1,2,1) 
> sapply(rowSel,function(i){mat[i,colSel[i]]}) 
[1] 1 6 3 

Answer 1

Daha kısa bir yol:

mat[cbind(rowSel, colSel)] 
#[1] 1 6 3 

Bu twocolumn matrisi tarafından indeksleme kullanır. İlk sütun satırın dizinini içerir, ikinci sütun sütunun dizinini içerir. İki sütun matrisinin her satırı, mat matrisinin bir öğesini endeksler.