np.nonzero()的用法【numpy小白】

numpy函数返回非零元素的位置。

返回值为元组， 两个值分别为两个维度， 包含了相应维度上非零元素的目录值。
 
比如：

n1=np.array([0,1,0,0,0,0,1,0,0,0,0,0,0,1])

n1.nonzero()

返回的是：

(array([ 1,  6, 13], dtype=int64),)

注意上面是一个yu元组
要获取里面的值，需要用 n1.nonzero()[0] 来获取。
 
原创文章
转载请注明出处：
http://30daydo.com/article/466