我有一个稀疏矩阵。
哪里
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),
....]
最佳答案
你可以试试这种拉链。主要思想是使用nonzero()方法
for i in range(len(A.nonzero()[0])):
print( (A.nonzero()[0][i],A.nonzero()[1][i]) )