如何根据搜索返回numpy ndarray的索引？

Question

我有一个numpy 2D数组，

import numpy as np
array1 = array([[ 1,  2,  1,  1],
   [ 2,  2,  2,  1],
   [ 1,  1,  1,  1],
   [1,  3, 1, 1],
   [1,  1, 1, 1]])

我想找到元素＆＃39; 3＆＃39;并知道它的位置。所以，我可以试试

condition = array1 == 3

然后找到参数，我可以尝试

np.argwhere(condition)

如何找到3所在的指数？

Answer 1

您还可以使用where，返回坐标元组：

In [34]: np.where(array1==3)
Out[34]: (array([3]), array([1]))