python - 访问 csr_matrix 的所有非零条目

Question

我有一个稀疏矩阵:

from scipy.sparse import csr_matrix
M=csr_matrix((5,5))
M[2,3]=4

我想迭代所有非零条目，例如:

for x,y,v in M.non_zero_entries:
   do_something()

我尝试理解 M.data、M.indices 和 M.indptr 的值

现在，在上面的示例中:

print (M.data) #outputs [4]
print (M.indices) #outputs [3]
print (M.indptr) #outputs [0,0,0,1,1,1]

如何从中提取非零记录？

Answer 1

您正在寻找的是 nonzero方法:

csr_matrix.nonzero()

Returns a tuple of arrays (row,col) containing the indices of thenon-zero elements of the matrix.

所以你的循环将是这样的:

for row, col in zip(*M.nonzero()):
    val = M[row, col]
    # do something
    print((row, col), val)