r - 如何对每行具有不同列位置的矩阵进行子集化？

Question

这个问题在这里已经有了答案:





Subset a matrix according to a columns vector

(2 个回答)


2年前关闭。




我想对每一行使用不同(但只有一个)列的矩阵进行子集化。那么适本地申请可以完成这项工作吗？但也许智能子集也可以工作，但我还没有找到解决方案。计算时间是一个问题 - 我有一个 for 循环的解决方案，但是在 RAM 中多次加载矩阵太慢了。
下面是一个例子:

矩阵 M 和向量 v 给定，

M<-matrix(1:15,nrow=5,ncol=3)

     [,1] [,2] [,3]
[1,]    1    6   11
[2,]    2    7   12
[3,]    3    8   13
[4,]    4    9   14
[5,]    5   10   15

v<-c(3,1,1,2,1)

解决办法是:

(11,2,3,9,5)

Answer 1

我们可以尝试行/列索引

M[cbind(1:nrow(M), v)]
#[1] 11  2  3  9  5