Как найти индекс значения в массиве 2d в Python?

Мне нужно выяснить, как я могу найти весь индекс значения в массиве 2d numpy.

Например, у меня есть следующий массив 2d:

([[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)].

Ответ 1

Вы можете использовать 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 может быть более эффективным способом генерации пар.

Ответ 2

Проблема с пониманием списка, которую вы указали, состоит в том, что она идет только на один уровень, вам нужно вложенное понимание списка:

a = [[1,0,1],[0,0,1], [1,1,0]]

>>> [(ix,iy) for ix, row in enumerate(a) for iy, i in enumerate(row) if i == 0]
[(0, 1), (1, 0), (1, 1), (2, 2)]

Если вы работаете с массивом numpy, лучше использовать встроенные функции, как предложено ajcr.

Ответ 3

Используя numpy, argwhere может быть лучшим решением:

import numpy as np

array = np.array([[1, 1, 0, 0],
                  [0, 0, 1, 1],
                  [0, 0, 0, 0]])

solutions = np.argwhere(array == 1)
print(solutions)

>>>
[[0 0]
 [0 1]
 [1 2]
 [1 3]]