r - 将which(,arr.ind=T)反馈回R矩阵的最佳方法？

Question

我已经提取了一些我想查看的元素的数组索引，如下所示:

mat = matrix(0,10,10)

arrInd = which(mat ==0,arr.ind = T)

然后我对该矩阵进行更多操作，最终得到一个向量或行 rowInd 和一个列向量 colInd。我希望我们将这些因数插入到另一个矩阵中，例如 mat2。但我似乎无法找到一种方法来做到这一点，而不需要循环或自己进行模块化算术计算。我意识到我可以采取类似的东西

mat2[rowInd*(colInd-1)+rowInd]

为了转换回一维索引。但由于 R 通常有内置函数来完成这类事情，我想知道是否有更简洁的方法来做到这一点？像 which(,arr.ind=T) 这样方便的数据操作函数有一个方便的逆函数，这似乎很自然。

我也尝试使用mat2[rowInd,colInd]，但这不起作用。

Answer 1

阅读R intro: indexing a matrix关于矩阵索引的使用。 which(, arr.ind = TRUE) 返回适合直接使用矩阵索引的两列矩阵。例如:

A <- matrix(c(1L,2L,2L,1L), 2)
iv <- which(A == 1L, arr.ind = TRUE)

#     row col
#[1,]   1   1
#[2,]   2   2

A[iv]
# [1] 1 1

如果您有另一个矩阵B，您想根据iv更新值，只需执行

B[iv] <- replacement

也许由于某种原因，您将行索引和列索引分离为 rowInd 和 colInd。在这种情况下，只需使用

cbind(rowInd, colInd)

作为索引矩阵。