如何在Python中的2d数组中找到值的索引？

如何在Python中的2d数组中找到值的索引？

您可以

np.where

用来返回x和y索引数组的元组，其中给定条件保存在数组中。

如果

a

是阵列名称：

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

如果要列出（x，y）对，则可以

zip

使用两个数组：

>>> zip(*np.where(a == 1))[(0, 0), (0, 1), (1, 2), (1, 3)]

或者，甚至更好的是，@ jme指出这

np.asarray(x).T

可能是生成配对的一种更有效的方法。