跨不同行和列的多个矩阵元素

Question

我试图在 R 中的矩阵或数据帧中跨列提取特定的行元素。一个简化的例子是：

dat<-rbind(rnorm(10, 0, 1), rnorm(10, 0, 1))

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

我希望在 dat 的列中返回 rown 指定的行索引。 我可以用一个循环来做到这一点：

for(i in 1:spec.num){print(dat[,i][rown[i]])}

但我知道必须有一种方法可以通过 apply 或类似的方法来做到这一点。 但是，我无法弄清楚如何将 rown 的元素合并到 apply 函数中。 如果 dat 是列列表，我也可以使用 mapply 轻松完成此操作，但这也不理想。

Answer 1

我们可以使用cbind创建行/列索引矩阵并使用它来对值进行子集化。

dat[cbind(rown, seq_len(ncol(dat)))]
#Equivalent call and maybe easy to understand
#dat[cbind(rown, 1:ncol(dat))]
#[1] -0.560 -0.230  1.559  0.111 -0.556  1.715  0.461 -1.265 -0.687 -0.473

数据

set.seed(123)
dat<-rbind(rnorm(10, 0, 1), rnorm(10, 0, 1))