r 中带有数组的子集矩阵

这可能是相当基本的，但我还没有找到一个简单的解决方案。

假设我有一个三维矩阵：

m <- array(seq_len(18),dim=c(3,3,2))

我想用索引数组对矩阵进行子集化：

idxrows <- c(1,2,3)
idxcols <- c(1,1,2)

获取位置上的数组(1,1),(2,1) and (3,2)， 那是：

       [,1] [,2] [,3]
[1,]    1    5    9
[2,]   10   14   18

我努力了m[idxrows,idxcols,]但没有任何运气。

有没有办法做到这一点（显然不使用 for 循环）？

不确定是否有任何简单的内置extract语法，但您可以使用以下方法解决此问题mapply:

mapply(function(i, j) m[i,j,], idxrows, idxcols)

#     [,1] [,2] [,3]
#[1,]    1    2    6
#[2,]   10   11   15

或者稍微复杂一点，创建一个索引矩阵，其列与原始数组的维度匹配：

thirdDim <- dim(m)[3]
index <- cbind(rep(idxrows, each = thirdDim), rep(idxcols, each = thirdDim), 1:thirdDim)
matrix(m[index], nrow = thirdDim)

#     [,1] [,2] [,3]
#[1,]    1    2    6
#[2,]   10   11   15