r - 通过索引向量更改多个矩阵元素

Question

我有一个矩阵

myMatrix <- matrix(data = 0, nrow = 4, ncol = 4)

     [,1] [,2] [,3] [,4]
[1,]    0    0    0    0
[2,]    0    0    0    0
[3,]    0    0    0    0
[4,]    0    0    0    0

我想更改特定值

myMatrix[1,1] <- 1
myMatrix[2,3] <- 1
myMatrix[4,4] <- 1

myMatrix

     [,1] [,2] [,3] [,4]
[1,]    1    0    0    0
[2,]    0    0    1    0
[3,]    0    0    0    0
[4,]    0    0    0    1

如果我有两个包含行和列索引的向量，我该如何高效/优雅地做到这一点:

rowIndexes <- c(1,2,4)
colIndexes <- c(1,3,4)

分配的值是常量(在本例中为 1 )。

我知道如何使用 for -loop，但这感觉效率低下。

Answer 1

我们可以cbind行/列索引，子集 myMatrix并将值赋值为 1

myMatrix[cbind(rowIndexes, colIndexes)] <- 1
myMatrix
#     [,1] [,2] [,3] [,4]
#[1,]    1    0    0    0
#[2,]    0    0    1    0
#[3,]    0    0    0    0
#[4,]    0    0    0    1