Numpy的argsort函数

函数声明如下:

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,其余同理。

你可能感兴趣的:(python)