[numpy]argsort函数

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
返回的是数组值从小到大的索引值
参数：
a为要排序的数组
axis：按哪一维进行排序
kind：排序算法的选择，有quicksort，mergesort，heapsort对于一维数组

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

二维数组

>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3],
[2, 2]])

>>> np.argsort(x, axis=0) #按列排序
array([[0, 1],
[1, 0]])

>>> np.argsort(x, axis=1) #按行排序
array([[0, 1],
[0, 1]])

np.argsort()[num]的形式
当num>=0时，np.argsort()[num]就可以理解为y[num];
当num<0时，np.argsort()[num]就是把数组y的元素反向输出.

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