Jak znaleźć indeks wartości w tablicy 2d w Pythonie?

Muszę się dowiedzieć, jak mogę znaleźć cały indeks wartości w tablicy numpy 2d.

Na przykład, mam następującą tablicę 2d: 

([[1 1 0 0],
  [0 0 1 1],
  [0 0 0 0]])
Muszę znaleźć indeks wszystkich 1 i 0.]} 
1: [(0, 0), (0, 1), (1, 2), (1, 3)]
0: [(0, 2), (0, 3), (1, 0), (1, 1), (the entire all row)]

Próbowałem tego ale nie daje mi wszystkich indeksów: 

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

W zasadzie daje mi tylko jeden indeks w każdym wierszu [(0, 0), (1, 2)].

Author: Alex Riley, 2014-11-27
Source

3 answers

Możesz użyć np.where zwrócenie krotki tablic indeksów x i y, w których dany warunek znajduje się w tablicy.

Jeśli a jest nazwą tablicy: 

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

Jeśli chcesz listę par (x, y), możesz zip dwie tablice: 

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

Lub, jeszcze lepiej, @jme zwraca uwagę, że np.asarray(x).T może być bardziej efektywnym sposobem generowania par.

 45
Author: Alex Riley,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2015-05-12 17:15:59

Problem ze zrozumieniem listy, który podałeś, polega na tym, że chodzi tylko o jeden poziom głębokości, potrzebujesz zagnieżdżonego zrozumienia listy: 

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)]

To powiedziawszy, jeśli pracujesz z tablicą numpy, lepiej jest użyć wbudowanych funkcji, zgodnie z sugestią ajcr.

 13
Author: Mike,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2014-11-27 16:51:01

Używając numpy, argwhere może być najlepszym rozwiązaniem: 

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]]
 7
Author: Guillaume Lebreton,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2019-09-24 20:18:10