scipy.sparse.find

scipy.sparse.find(A)[源代码]

返回矩阵的非零元素的索引和值

参数
A稠密或稀疏矩阵

需要非零元素的矩阵。

退货
(I、J、V)数组的元组

I、J和V包含非零矩阵条目的行索引、列索引和值。

示例

>>> from scipy.sparse import csr_matrix, find
>>> A = csr_matrix([[7.0, 8.0, 0],[0, 0, 9.0]])
>>> find(A)
(array([0, 0, 1], dtype=int32), array([0, 1, 2], dtype=int32), array([ 7.,  8.,  9.]))