scipy.sparse.

find

scipy.sparse.find(A)

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

参数:

A稠密或稀疏数组或矩阵

希望获取其非零元素的矩阵。

返回:

(I,J,V)数组元组

I、J 和 V 分别包含非零项的列索引、行索引和值。

示例

在浏览器中尝试！

>>> from scipy.sparse import csr_array, find
>>> A = csr_array([[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.]))

返回在新标签中打开