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

我知道有一种方法可以返回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)

到

---

你想要这样的东西吗：

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

---

请参阅：，

您想要这样的东西吗：

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

---

请参阅：，

是。我无法相信为什么我没有从谷歌得到正确的答案。是的。我无法相信为什么我没有从谷歌得到正确的答案。