返回數組中每個非零元素的索引

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)

至

[[1, 2, ...,6],
[1,2,...6,8],
...
...
]

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]