np.argsort()用法

np.argsort()

将数组里的元素默认按照从小到大排序，取出相应的索引进行输出

import numpy as np
a = np.array([1,-2,3,5,4])
x = a.argsort()
>> [1,0,2,4,3]
y = np.argsprt(-a)  # 相当于返回元素从大到小排序的索引
>> [3,4,2,0,1]

应用示例：
将一个数组按照每一行的第一个元素的大小重新排序（从小到大）

import numpy as np
a = np.array([[1,-2,3,5,4],[0,1,2,3,4],[-2,1,1,1,1]])
p = a[np.argsort(a[:,0])]
>> [[-2,1,1,1,1],[0,1,2,3,4],[1,-2,3,5,4]]