在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