用法: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 |
行坐标:nonzero(a)[0] = array([0,0,1,2,2])
列坐标:nonzero(a)[1] = array([0,1,2,0,2])