numpy——nonzero

用法：nonzero(a)

作用：返回數組a中的非零元素的座標

返回值：一個二元tuple值，第一個值爲行座標，第二值爲列座標

例子：

>>> from numpy import *
>>> a = [[1,1,0],[0,0,1],[1,0,1]]
>>> nonzero(a)
(array([0, 0, 1, 2, 2]), array([0, 1, 2, 0, 2]))

註解：

a = 

1	1	0
0	0	1
1	0	1

所以，非零元素的座標爲(0,0),(0,1),(1,2),(2,0),(2,2)五個非零元素。

行座標：nonzero(a)[0] = array([0,0,1,2,2])

列座標：nonzero(a)[1] = array([0,1,2,0,2])