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])

转载于:https://www.cnblogs.com/ritchiewang/p/5767397.html