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

Elemente mithilfe der Integer-Array-Indizierung extrahieren

Bei der Arbeit mit mehrdimensionalen Arrays ist es häufig erforderlich, bestimmte Elemente anhand von Indizes zu extrahieren. In NumPy können Sie dazu verschiedene Techniken nutzen. Eine solche Methode ist die Verwendung der ganzzahligen Array-Indizierung.

Betrachten Sie das folgende Beispiel:

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

Unser Ziel ist es, ein neues Array C zu erstellen, das Elemente aus A enthält, wobei der Zeilenindex für jedes Element angegeben ist Das Element wird durch A.shape[0] und der Spaltenindex durch die raveled-Version von B angegeben. Mit anderen Worten, C sollte sein:

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

Ein Ansatz besteht darin, die Integer-Array-Indizierung wie folgt zu verwenden:

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

Dieser Ansatz verwendet die Arange-Funktion, um einen Bereich von Indizes für die Zeilen von A zu generieren, und kombiniert ihn dann mit der ravelierten Version von B, um die Spaltenindizes zu erstellen. Das Ergebnis ist ein neues Array, das die gewünschten Elemente enthält.

# Sample run
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]

Es ist wichtig zu beachten, dass, wenn B ein 1D-Array oder ein ist Liste können Sie den Reduzierungsvorgang mit .ravel() überspringen.

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