python - 向稀疏矩阵中的非零元素添加值

Question

我有一个稀疏矩阵，我想将所有非零元素的值加一。但是，我无法弄清楚。有没有办法在 python 中使用标准包来做到这一点？任何帮助将不胜感激。

标签： pythonmachine-learningsparse-matrix

Answer 1

我无法评论它的性能，但你可以（Scipy 1.1.0）；

>>> from scipy.sparse import csr_matrix
>>> a = csr_matrix([[0, 2, 0], [1, 0, 0]])
>>> print(a)
(0, 1)        2
(1, 0)        1
>>> a[a.nonzero()] = a[a.nonzero()] + 1
>>> print(a)
(0, 1)        3
(1, 0)        2