Verwenden der Integer-Array-Indizierung zum Extrahieren von Elementen basierend auf sekundären Array-Indizes

In dem gegebenen Szenario besteht das Ziel darin, bestimmte Elemente aus einem abzurufen Array A mithilfe von Indizes, die in einem zweiten Array B angegeben sind. Anstatt sich auf np.take oder np.choose zu verlassen, besteht ein einfacherer Ansatz darin, die Integer-Array-Indizierung von NumPy zu verwenden:

A[np.arange(A.shape[0]),B.ravel()]

So erzielt dieser Code das gewünschte Ergebnis:

• np.arange(A.shape[0]) erstellt ein Array von Indizes von 0 bis (A.shape[0] - 1), die die Zeilen von A darstellen.
• B.ravel() glättet B in ein eindimensionales Array und stellt so sicher, dass es die gleiche Anzahl von Elementen hat als Zeilen in A.
• Die Indexierungsoperation A[...] ruft Elemente von A unter Verwendung der Zeilenindizes und Spaltenindizes von B ab.

Dieser Ansatz ist besonders nützlich, wenn B ist ein 1D-Array oder eine Liste von Spaltenindizes. Durch das Überspringen der Reduzierungsoperation wird der Code noch einfacher:

A[np.arange(A.shape[0]),B]

Beispiel:

A = np.array([[0, 1], [2, 3], [4, 5]])
B = np.array([1, 0, 1])

result = A[np.arange(A.shape[0]), B]

print(result)
# Output: [1, 2, 5]