python中的argsort获取的是排好序的位置上原来元素的index

argsort 是许多编程语言中数组处理库的一个常见函数，用于返回数组中元素的索引，这些索引将数组的元素排序到升序位置。换句话说，argsort 不直接返回排序后的数组，而是返回原始数组中元素应该被放置的索引，以便生成排序后的数组。

在 Python 的 NumPy 库中，numpy.argsort 是一个特别有用的函数。以下是一个简单的例子：

import numpy as np 


# 创建一个数组 
               # 3 1 4 1 5
               # 0 1 2 3 4 index
               # 1 1 3 4 5 sorted
arr = np.array([3, 1, 4, 1, 5]) 


# 使用 argsort 获取排序索引 
sorted_indices = np.argsort(arr) # 1 3 0 2 4   这是原来数字下的坐标，然后放在现在的位置是排好序的位置原来元素的index


# 打印排序索引 
print(sorted_indices) # 输出: 1 3 0 2 4  


# 使用这些索引对原始数组进行排序 
sorted_arr = arr[sorted_indices] # 改变原来数组的index  sorted_arr[0] = arr[1] = 1 sorted_arr[1] = arr[3] = 1  sorted_arr[2] = arr[0] = 3


# 打印排序后的数组 
print(sorted_arr) # 输出: [1 1 3 4 5]

在上面的例子中，np.argsort(arr) 返回了一个数组，该数组的元素是 arr 中元素应该被放置的索引，以便得到一个升序排列的数组。

然后，我们使用这些索引对原始数组 arr 进行排序，得到排序后的数组 sorted_arr。其实，相当于改变原来数组的index  即sorted_arr[0] = arr[1] = 1 sorted_arr[1] = arr[3] = 1  sorted_arr[2] = arr[0] = 3

注意，argsort 默认返回升序索引。如果你想要降序索引，你可以对返回的索引数组进行反转，或者对原始数组进行降序排序并获取其索引。例如：

# 获取降序索引 
descending_indices = np.argsort(arr)[::-1] 
print(descending_indices) # 输出: [5，4，3，1，1]