Come trovare l'indice di un valore in array 2d in Python?

Question

Ho bisogno di capire come posso trovare tutto l'indice di un valore in una matrice numdy 2d.

Per esempio, ho il seguente array 2d:

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

ho bisogno di trovare l'indice di tutte le 1 e 0.

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

Ho provato questo ma non mi dà tutti gli indici:

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

Fondamentalmente, mi dà solo una dell'indice in ogni riga [(0, 0), (1, 2)].

Answer 1

È possibile utilizzare np.where per restituire una tupla di matrici di indici xey in cui una determinata condizione contiene una matrice.

Se a è il nome della matrice:

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

Se si desidera un elenco di (x, y) le coppie, si potrebbe zip i due array:

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

O, ancora meglio, @jme sottolinea che np.asarray(x).T può essere un modo più efficiente per generare le coppie.

Answer 2

Il problema con la comprensione lista che hai fornito è che si va un solo livello, è necessario un elenco nidificato di comprensione:

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

Detto questo, se si sta lavorando con una serie NumPy, è meglio utilizzare le funzioni integrate come suggerito da ajcr.