scipy.sparse.dia_array.

nonzero#

dia_array.nonzero()[源代码]#

数组/矩阵的非零索引。

返回一个数组元组 (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], dtype=int32), array([0, 1, 2, 0, 2], dtype=int32))