函数声明如下:
numpy.
argsort
(a, axis=-1, kind='quicksort', order=None)
参数
a:马上要排序的numpy.array
axis:按照哪个维度来排序,默认是-1。如,对于二维数组来说,axis=0,就是按列为单位排序,aixs=1,就是按行来。
kind和order本人现阶段不用更改,不管。
返回值是 将要被排序的a的下标数组。意思是不是返回排好序的数组,而是返回排好序数组的每个元素在原数组a中的下标位置。
比如:
a=np.arrary([5,6,1,4,0])
b=np.argsort(a)
那么这个时候的b=[4,2,3,0,1],因为排好序的数组是[0,1,4,5,6],0在a中下标是4,1在a中下标是2,其余同理。