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

时间:2018-04-04 13:14:23

标签: r matrix sparse-matrix

说我有一个稀疏矩阵:

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在技术上不再是压缩的稀疏矩阵,因为某些&#34;非零&#34;元素现在实际上为零。

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中所示),或循环遍历元素手动

1 个答案:

答案 0 :(得分:2)

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

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 . . .