numpy nonzero

numpy.nonzero(a): 返回的是a中非0元素的索引的元组，经常可以用a[nonzero(a)]得到a中非0元素

>>> x = np.eye(3)
>>> x
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
>>> np.nonzero(x)
(array([0, 1, 2]), array([0, 1, 2]))

上面例子说明a数组非0元素的下标为（0,0） （1,1） （2,2）

>>> x[np.nonzero(x)]
array([ 1.,  1.,  1.])
>>> np