scipy.sparse.dia_matrix.

nonzero#

dia_matrix.nonzero()[source]#

数组/矩阵的非零元素索引。

返回包含数组非零元素索引的数组元组 (row,col)。

示例

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