在python中列出稀疏矩阵中的非零元素

Question

如何以简单的一行代码（并且快速！）列出 a 的所有非零元素csr_matrix？

我正在使用这段代码：

edges_list = list([tuple(row) for row in np.transpose(A.nonzero())])
weight_list = [A[e] for e in edges_list]

但执行起来需要相当长的时间。

Answer 1

对于规范形式的 CSR 矩阵，直接访问数据数组：

A.data

但请注意，非规范形式的矩阵可能在其表示中包含显式零或重复条目，这需要特殊处理。例如，

# Merge duplicates and remove explicit zeros. Both operations modify A.
# We sum duplicates first because they might sum to zero - for example,
# if a 5 and a -5 are in the same spot, we have to sum them to 0 and then remove the 0.
A.sum_duplicates()
A.eliminate_zeros()

# Now use A.data
do_whatever_with(A.data)

Answer 2

只需使用A.data

In [16]: from scipy.sparse import csr_matrix

In [17]: A = csr_matrix([[1,0,0],[0,2,0]])

In [18]: A.data
Out[18]: array([1, 2])

如果稀疏矩阵已被修改或者为了安全起见，您应该使用：A.eliminate_zeros()

In [19]: A[0,0] = 0

In [20]: A.data
Out[20]: array([0, 2])

In [21]: A.eliminate_zeros()

In [22]: A.data
Out[22]: array([2])