Ein kurzer Überblick über die Verwendung der argsort()-Funktion in Python

Dieser Artikel stellt hauptsächlich die Verwendungsbeispiele der argsort()-Funktion in Python vor und stellt die Verwendung der argsort()-Funktion im Detail vor. Interessierte Freunde können darauf verweisen.

Da ich Python verwenden möchte, um viele Bilder mit dem trainierten Caffemodel zu klassifizieren, bin ich auf die Argsort-Funktion gestoßen, also habe ich die relevante Literatur überprüft und sie selbst in der Python-Umgebung getestet Um es später nicht zu vergessen, habe ich es aufgeschrieben, um mein Verständnis zu vertiefen. (Ps: Ich bin auch ein Python-Neuling, daher ist mein Verständnis möglicherweise oberflächlich)

1 Definieren Sie zuerst Array-Daten

import numpy as np
x=np.array([1,4,3,-1,6,9])

2 die argsort()-Funktion Was ist:

x.argsort()

Die Ausgabe ist definiert als y=array([3,0,2,1,4,5]).

Wir haben festgestellt, dass die Funktion argsort() die Elemente in x von klein nach groß anordnet, ihren entsprechenden Index extrahiert und ihn dann nach y ausgibt. Beispiel: x[3]=-1 ist das kleinste, also y[0]=3, x[5]=9 ist das größte, also y[5]=5.

Das Obige ist nicht schwer zu verstehen. Wenn Sie damit nicht vertraut sind, können Sie es selbst in der Python-Umgebung ausprobieren.

3. Da ich im Programm auf ein ähnliches Formular wie np.argsort()[num] gestoßen bin, konnte ich es immer noch nicht verstehen, also bin ich in die Python-Umgebung gegangen und habe es selbst ausprobiert:

ps: Die Zahl hier ist absolut. Der Wert ist kleiner oder gleich der Anzahl der Elemente in x

, wenn num>=0时，np.argsort()[num]就可以理解为y[num];

, wenn num<0时，np.argsort()[num]就是把数组y的元素反向输出, zum Beispiel np.argsort()[- 1] gibt den Index aus, der dem Maximalwert in x entspricht, np.argsort()[-2] gibt den Index aus, der dem zweitgrößten Wert in x entspricht, und so weiter. .

Nur ​​durch intuitive Experimente können Sie den Effekt erkennen. Hier ist meine Überprüfung anhand des obigen Beispiels:

Dies ist die Ausgabe, wenn num ein negativer Wert ist .

Dies ist die Ausgabe, wenn num>=0.

Das obige ist der detaillierte Inhalt vonEin kurzer Überblick über die Verwendung der argsort()-Funktion in Python. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!