numpy argsort使用

对于一个无序列表

lst = [3, 1, 4, 1, 5, 9, 2]

使用numpy的argsort函数进行升序排序，得到排序后的索引

sorted_indices = np.argsort(lst)
# 排序后的列表 [1, 1, 2, 3, 4, 5, 9]
# 返回：array([1, 3, 6, 0, 2, 4, 5])
# 分别对应：
# 原列表中的第0个元素3，对应在排序后的列表的第3位，因此在返回的列表中的第3位上写0，代表该元素在原列表lst中的索引
# 类似的， 原列表中的第1个元素1，对应在排序后的列表的第0位，因此在返回的列表中的第0位上写1
# 原列表中的第2个元素4，对应在排序后的列表的第4位，因此在返回的列表中的第4位上写2

如果需要得到该无序列表中的每个元素与其在排序后列表中的元素位置对应，则需要：

sorted_indices = np.argsort(np.argsort(lst))
# 排序后的列表 [1, 1, 2, 3, 4, 5, 9]
# 返回：array([3, 0, 4, 1, 5, 6, 2])
# 原列表中的第0个元素3，对应在排序后的列表的第3位，因此在返回的列表中的第0位上写3，代表该元素在原列表lst中升序排序的序号
# 类似的，原列表中的第1个元素1，对应在排序后的列表的第0位，因此在返回的列表中的第0位上写0，即是原始列表中最小的元素
# 原列表中的第2个元素4，对应在排序后的列表的第4位，因此在返回的列表中的第0位上写4

如果需要降序，则将原列表前面加一个负号即可，np.argsort(-lst).