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

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]