python - 将 Scipy 稀疏矩阵转换为元组

Question

我有一个稀疏矩阵让我们说 A。

在哪里

type(A) 
scipy.sparse.csr.csr_matrix

和一个

<100x100 sparse matrix of type '<class 'numpy.int64'>'
with 198 stored elements in Compressed Sparse Row format>

得到如下结果

(0, 1)  1
(0, 0)  1
(0, 2)  1
(0, 3)  1
(0, 4)  1
(0, 5)  1
(0, 6)  1
....

表示矩阵A中的非零元素。(代码如下)

for a in A:
  print(a)

如何将其转换为如下数据结构:

[(0,1),
(0,0),
(0,2),
....]

Answer 1

你可以试试这种zip。主要思想是使用nonzero()方法

for i in range(len(A.nonzero()[0])):
     print( (A.nonzero()[0][i],A.nonzero()[1][i]) )