Wie kann ich den ersten Index eines Elements in einem NumPy-Array finden?

Den ersten Index eines Elements in einem NumPy-Array finden

In Python-Listen können wir mit der Methode index() den ersten Index abrufen Vorkommen eines Elements. Kann NumPy eine ähnliche Funktionalität für Arrays bereitstellen?

Antwort

Ja, NumPy bietet eine praktische Methode zum Auffinden des ersten Index eines Elements in einem Array. Sie können die Funktion np.where wie folgt verwenden:

import numpy as np

array = np.array([1, 2, 3])
item = 2

itemindex = np.where(array == item)

Die Funktion np.where gibt ein Tupel zurück, das die Zeilen- und Spaltenindizes enthält, in denen das Element gefunden wird.

Beispiel

Wenn Ihr Array zweidimensional ist und das Element an zwei Stellen enthält, dann:

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

Dies ermöglicht Ihnen den einfachen Zugriff auf das gewünschte Element basierend auf seinem ersten Vorkommen.

Das obige ist der detaillierte Inhalt vonWie kann ich den ersten Index eines Elements in einem NumPy-Array finden?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!