r - 如何提取与列列表对应的矩阵元素

Question

使用应用函数似乎应该非常简单，但我发现自己在努力解决这个问题。

我有一个数据矩阵(数据框也可以):

u <- matrix(sample(seq(4),20,T),5,4)
u
     [,1] [,2] [,3] [,4]
[1,]    1    2    4    2
[2,]    4    3    2    2
[3,]    3    3    3    1
[4,]    3    2    4    4
[5,]    4    1    3    4

假设我只想使用元素(如 excel 中的间接元素)第 j 列从每一行中选择相应的列值。

例如给定 col(j) = 3

row 1 would get element corresponding to row=1,col(j=3)=4 and return 2 (row(1):col(4))

row 2 would get elementcorresponding to row=2,col(j=3)=2 and return 3 (row(2):col(2))

...

row5 would get element corresponding to row=5,col(j=3)=3 and return 3(row(5),col(3))

I end up with a vector of those values v<-c(4,2,...3)

Answer 1

您可以使用矩阵索引:

i <- seq_len(nrow(u))
j <- u[, 3]
u[cbind(i, j)]