in Array of Arrays effizient den index ermitteln

[Erhy]

Hallo!

wie kann ich den Index eines eingebetteten arrays ermitteln?

z.B.

Code: Alles auswählen

bigArray = np.array([[1, 2, 3], [3, 4, 5], [5, 6, 7]])
thisArray = np.array([3, 4, 5])

np.asarray(np.where(bigArray == thisArray))

ergibt

Code: Alles auswählen

array([[1, 1, 1],
       [0, 1, 2]], dtype=int64)

anstatt

Code: Alles auswählen

array([[1]])

Danke für Eure Antwort
Erhy

[__blackjack__]

Code: Alles auswählen

In [502]: rows                                                                  
Out[502]: 
array([[1, 2, 3],
       [3, 4, 5],
       [5, 6, 7]])

In [503]: needle                                                                
Out[503]: array([3, 4, 5])

In [504]: rows == needle                                                        
Out[504]: 
array([[False, False, False],
       [ True,  True,  True],
       [False, False, False]])

In [505]: (rows == needle).all(axis=1)                                          
Out[505]: array([False,  True, False])

In [506]: (rows == needle).all(axis=1).nonzero()                                
Out[506]: (array([1]),)

[Erhy]

danke für die schnelle Antwort.

die letzte Zeile habe ich noch modifiziert und hinzugefügt:

Code: Alles auswählen

d = (rows == needle).all(axis=1).nonzero()
arrInDuple = d[0]
i = arrInDuple[0]  # index of searchrd array in big array