numpy.argsort（）详细说明及例子

1 numpy.argsort（）函数作用

	将传入的列表a中的元素按从小到大进行排列，然后返回一个列表y，y中元素即为排列后原列表a中对应数字的索引号，下面举个例子理解

import numpy as np
a = np.array([5,1,4,3,9])
y = np.argsort(a)
print(y.shape)#打印返回值的数据大小
print(y)

2 代码输出结果如下

3 分析

由此可知，返回值y为一个列表，里面的值正是先按从小到大排列后的原输入索引，a按从小到大排序为 1，3，4，5，9，对应的位置为a[1],a[3],a[2],a[0],a[4]，索引值为1,3，2,0，4，正好就是y的值

4拓展

yi = np.argsort(a)[i] 

也就相当于

yi = y[i]

返回的值为y中的对应元素，索引从0开始，表示第一个数值，1表示第二个数值，以此类推，同时索引为-1时表示最后一个数值，-2表示倒数第二个数值，以此类推