Python 如何获取numpy数组中给定项的列和行索引（y和x）

例如，我有一个3x3 numpy数组，如下所示

a=numpy.array([3,2,1],
              [2,5,6],
              [4,7,8])

如果给定的值是8，那么我应该得到类似于[3,3]的值 如果给定的值是2，那么我应该得到[0,1]和[1,0]，因为数组中有两个'2'

我知道这对于循环来说很简单，但我想知道是否有一种有效的Pythonic方法或任何花哨的numpy函数来解决它，只需要最少的代码行。谢谢

您可以使用：

---

这将为您提供元组中数组的索引

np.where(a == 3) # (array([0]), array([0]))
np.where(a == 2) # (array([0, 1]), array([1, 0]))

np.where(a == 3) # (array([0]), array([0]))
np.where(a == 2) # (array([0, 1]), array([1, 0]))