返回数组中每个非零元素的索引

Question

我知道在python中有一种方法可以返回数组中最大元素的索引：numpy.argmax()。有没有办法返回每个非零元素的索引？在

例如

array([[ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  1.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       ..., 
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.]], dtype=float32)

到

^{pr2}$

Answer 1

你想要这样的东西吗：

x = np.asarray([0, 1, 2, 3, 0, 1])
In [129]: np.nonzero(x)
Out[129]: (array([1, 2, 3, 5]),)

见：[1]，[2]，[3]