Wenn Sie argsort für ndarray verwenden, wird der Index in aufsteigender Reihenfolge sortiert zurückgegeben.
>>> a=np.array([1,4,2,5,3])
>>> print(a)
[1 4 2 5 3]
>>> np.argsort(a)
array([0, 2, 4, 1, 3])
Von den Werten im ndarray ... Da der Mindestwert 1 ist, ist sein Index 0, Der nächstkleinere Wert ist 2, sein Index ist also 2 Der nächstkleinere Wert ist 3, sein Index ist also 4 Das Ergebnis ist wie ...
Wenn Sie den Index des drittkleinsten Werts wissen möchten, können Sie wie folgt vorgehen.
>>> np.argsort(a)[2]
4
Einmal bestätigt, ist es wie folgt
>>> a[4]
3
Die Standardeinstellung ist aufsteigend. Verwenden Sie daher Slices, wenn Sie absteigen möchten.
>>> np.argsort(a)[::-1]
array([3, 1, 4, 2, 0])
Recommended Posts