argsort
不直接返回排序后的数组,而是返回原始数组中元素应该被放置的索引,以便生成排序后的数组。
import numpy as np # 创建一个数组
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]) # 使用 argsort 获取排序索引
sorted_indices = np.argsort(arr) # 打印排序索引
print(sorted_indices) # 输出: [1 6 0 3 2 9 4 7 8 5 10] # 使用这些索引对原始数组进行排序
sorted_arr = arr[sorted_indices] # 打印排序后的数组
print(sorted_arr) # 输出: [1 1 2 3 3 4 5 5 5 6 9]