如何在Python中找到二维数组中某个值的索引？

我需要弄清楚如何找到 2d numpy 数组中某个值的所有索引。

例如，我有以下二维数组：

([[1 1 0 0],
  [0 0 1 1],
  [0 0 0 0]])

我需要找到所有 1 和 0 的索引。

1: [(0, 0), (0, 1), (1, 2), (1, 3)]
0: [(0, 2), (0, 3), (1, 0), (1, 1), (the entire all row)]

我尝试了这个，但它没有给我所有索引：

t = [(index, row.index(1)) for index, row in enumerate(x) if 1 in row]

基本上，它只给我每行中的一个索引[(0, 0), (1, 2)].

您可以使用np.where http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html返回 x 和 y 索引数组的元组，其中给定条件在数组中成立。

If a是你的数组的名称：

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

如果你想要一个 (x, y) 对的列表，你可以zip两个数组：

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

或者，更好的是，@jme 指出np.asarray(x).T可以是生成对的更有效的方法。