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]