Python SciPy spmatrix.nonzero用法及代碼示例

本文簡要介紹 python 語言中 scipy.sparse.spmatrix.nonzero 的用法。

用法:

spmatrix.nonzero()

非零索引

返回包含矩陣非零元素索引的數組元組 (row,col)。

例子：

>>> from scipy.sparse import csr_matrix
>>> A = csr_matrix([[1,2,0],[0,0,3],[4,0,5]])
>>> A.nonzero()
(array([0, 0, 1, 2, 2]), array([0, 1, 2, 0, 2]))