python nonzero()函数的用法

作用：找到非零元素位置
eg

from numpy import *
a = mat([[1,1,0],[1,1,0],[1,0,3]])
print(a)

输出

[[1 1 0]
 [1 1 0]
 [1 0 3]]

print(a.nonzero())

输出：

(array([0, 0, 1, 1, 2, 2]), array([0, 1, 0, 1, 0, 2]))

第一个向量表示行，第二个表示列

print(a[a.nonzero()])  

输出

[[1 1 1 1 1 3]]