有效地将稀疏矩阵的元素设置为 0

Efficiently setting elements of a sparse matrix to 0

假设我有一个稀疏矩阵:

library(Matrix)
m <- Matrix(rpois(100, 1), 10, 10, sparse=TRUE)
m
#10 x 10 sparse Matrix of class "dgCMatrix"

# [1,] . . 2 3 1 1 1 . 3 1
# [2,] 1 . 1 . 1 . 2 . . .
# [3,] 1 4 1 . . . . 5 1 1
# [4,] . 1 1 . . . 4 1 1 1
# [5,] 1 . 4 . 1 . . 1 . 2
# [6,] 1 1 2 1 . 1 . 1 2 1
# [7,] 4 2 1 1 . . . 1 1 3
# [8,] 2 2 1 1 1 1 . 1 1 2
# [9,] 2 1 1 . 3 . 2 . . 1
#[10,] . 2 1 1 . . . 1 3 3

我想将所有小于或等于 1 的元素设置为零。这样做安全吗?请注意,m2 在技术上不再是压缩的稀疏矩阵,因为某些 "nonzero" 元素现在实际上为零。

m2@x[m2@x < 2] <- 0
m2
#10 x 10 sparse Matrix of class "dgCMatrix"

# [1,] . . 2 3 0 0 0 . 3 0
# [2,] 0 . 0 . 0 . 2 . . .
# [3,] 0 4 0 . . . . 5 0 0
# [4,] . 0 0 . . . 4 0 0 0
# [5,] 0 . 4 . 0 . . 0 . 2
# [6,] 0 0 2 0 . 0 . 0 2 0
# [7,] 4 2 0 0 . . . 0 0 3
# [8,] 2 2 0 0 0 0 . 0 0 2
# [9,] 2 0 0 . 3 . 2 . . 0
#[10,] . 2 0 0 . . . 0 3 3

特别是,我不想创建一个巨大的指标矩阵来给出要改变的元素的位置(如m[m < 2] <- 0),或者循环手动通过元素。

我没有发现您的方法有任何不安全之处,甚至可能会被推荐。如果你想得到一个完全稀疏的矩阵,你可以这样做:

drop0(m2)
#10 x 10 sparse Matrix of class "dgCMatrix"
#                         
# [1,] 2 . . . . . . . . .
# [2,] . . . 2 . . . . . .
# [3,] . . 2 . . . . . 2 .
# [4,] 2 2 . . 4 . . 2 2 3
# [5,] . 4 . 3 . 3 . . . .
# [6,] 2 . . . . 3 2 . . .
# [7,] . . . . . . 2 . . 2
# [8,] . . . . . . 2 . . 4
# [9,] 2 . . 2 . . . . . 2
#[10,] . . . . . . 2 . . .