Lorsque je fais des opérations de liste, je veux souvent un index jusqu'à la nième valeur la plus élevée. Jusqu'à présent, je l'ai malaxé de manière appropriée, mais je savais que l'utilisation de numpy le rendrait plus intelligent, alors j'ai pris une note pour moi-même.
a=[1,5,4,3,2]
Quand il y avait une liste
>>> A=np.array(a)
>>> A.argsort()
array([0, 4, 3, 2, 1])
Vous pouvez obtenir l'index par ordre croissant en utilisant argsort () pour et ndarray. Si vous voulez descendre, utilisez des tranches
>>> A.argsort()[::-1]
array([1, 2, 3, 4, 0])
Peut être obtenu
Recommended Posts