r - 将值分配给 R 中的矩阵 - 值进入不同列的每一行

Question

nr <- 2
nc <- 4
rxc <- nr*nc
m1 <- Matrix(0, nrow = nr, ncol = rxc, sparse = TRUE)
col <- 0
for(r in 1:nr){
  m1[r,((r-1)*nc+1) :(((r-1)*nc+1)+3)] <- 1
  col <- col+1
}

期望的输出:

[1,] 1 1 1 1 . . . .
[2,] . . . .    1 1 1 1

这可行，但它非常慢，因为我必须对 1000 万行执行此操作。在 R 中有更好的方法吗？

Answer 1

使用矩阵索引，它使用行/列的两列矩阵来确定更改了哪些单元格:

m1[cbind(rep(1:nr,each=nc),1:(nr*nc))] <- 1
m1
#2 x 8 sparse Matrix of class "dgCMatrix"
#                    
#[1,] 1 1 1 1 . . . .
#[2,] . . . . 1 1 1 1