Wie arbeite ich mit numpy.where?

Question

Ich möchte Indizes des Arrays wie x = np.array([[1, 1, 1], [2, 2, 2]]) wo Elemente gleich y = np.array([1, 1, 1]). Also tat ich das:

In: np.where(x == y) 
Out: (array([0, 0, 0]), array([0, 1, 2])) 

Es ist die richtige Antwort. Aber ich erwarte, nur den Index 0 zu bekommen, weil das Nullelement von x gleich y ist.

Answer 1

Sie müssen (x == y).all(axis=1) verwenden, um das Vergleichsergebnis über axis=1 zuerst zu reduzieren, das heißt alle Elemente sind gleich:

np.where((x == y).all(axis=1))[0] 
# array([0])