Wie extrahiere ich Elemente aus einem NumPy-Array mithilfe der Integer-Array-Indizierung?

Indizierte Extraktion mit einem NumPy-Array

Bei der Arbeit mit mehrdimensionalen Arrays in NumPy ist es oft notwendig, bestimmte Elemente basierend auf Indizes zu extrahieren. In diesem Zusammenhang können bestimmte Szenarien auftreten, in denen ein Array (B) als Index für die zweite Dimension eines anderen Arrays (A) dient.

Betrachten Sie ein Beispiel mit zwei Arrays A und B:

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

Unser Ziel ist es, ein Element aus jeder Zeile von A zu extrahieren, wobei der Elementindex durch das entsprechende Element in B bestimmt wird. Die gewünschte Ausgabe ist:

C = np.array([[1],
              [2],
              [5]])

Lösung mit Integer-Array-Indizierung:

Anstatt B zu senden, können wir NumPys rein Integer-Array-Indizierung verwenden:

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

Hier erstellt np.arange(A.shape[0]) ein Array von Indizes entsprechend der Anzahl von Zeilen in A, während B.ravel() B in ein 1D-Array reduziert. Indem wir diese kombinieren, können wir die entsprechenden Elemente aus A extrahieren.

Beispiellauf:

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

Ausgabe:

[[0 1]
 [2 3]
 [4 5]]
[[1]
 [0]
 [1]]
[1 2 5]

Dieser Ansatz bietet eine präzise und effiziente Lösung für unser Extraktionsproblem.

Das obige ist der detaillierte Inhalt vonWie extrahiere ich Elemente aus einem NumPy-Array mithilfe der Integer-Array-Indizierung?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!