矩陣索引ř

Question

我正在使用此代碼試圖索引的矩陣R：

test <- matrix(0, nrow = 10, ncol = 2) 
test[1:10, 1] <- 1:10 
test[1:10, 2] <- 11:20 

index <- c(1,2,1,1,2,1,1,2,1,2) 

answer <- test[ , index] 

我接收10×10矩陣。但我想收到一個載體。

[1] 1 12 3 4 15 6 7 18 9 20

任何想法？

編輯： 此外，如何使用排除索引（例如，-index）從矩陣中排除值。

[1] 11 2 13 14 5 16 17 8 19 10

Answer 1

我們需要row/column索引由相應的位置提取的元素。爲此，我們可以將cbind行序列與'索引'列並獲取元素。

test[cbind(seq_len(nrow(test)), index)] 
#[1] 1 12 3 4 15 6 7 18 9 20