numpy: argsort 的使用原理

numpy: argsort 的使用原理

单词解析

argsort： arg --> argument 论据；sort --> 排序
元素的论据，支撑就是下标，所以argsort的意思是下标排序

我们来看一个

实例

dist = np.array([3,1,2,4])
--> array([ 3,1,2,4])
np.sort(dist)  # 元素排序
--> array([1,2,3,4])  
np.argsort(dist)  # 元素一一对应后，列出其对应下标
--> array([1, 2, 0, 3], dtype=int64)   

解析：

aggsort 的使用原理是

1、先明确原始数据的下标
array([3,1,2,4]) 对应的下标是 0,1,2,3 即；
3 <----> 0
1 <----> 1
2 <----> 2
4 <----> 3

2、将原始数据排序
结果为array([1,2,3,4])

3、排序后的元素下标与原始数据一一对应
将其对应的下标列出即可得到结果
1 <----> 1
2 <----> 2
3 <----> 0
4 <----> 3

勤思考，多动手。
– Edwin