Python 数组同步排序:使用 argsort 函数
可以使用 numpy 库中的 argsort 函数来对一个数组进行排序,并返回排序后的索引值。然后根据这个索引值,可以同步调整另一个数组的元素。
示例代码如下:
import numpy as np
# 原始数组
arr1 = np.array([4, 2, 1, 3])
arr2 = np.array([10, 20, 30, 40])
# 对 arr1 进行排序,并返回排序后的索引值
sorted_indices = np.argsort(arr1)
# 根据排序后的索引值,同步调整 arr2 的元素
sorted_arr2 = arr2[sorted_indices]
print(sorted_arr2)
输出结果为:
[30 20 40 10]
在这个示例中,原始的 arr1 为 [4, 2, 1, 3],arr2 为 [10, 20, 30, 40]。对 arr1 进行从小到大排序后得到 [1, 2, 3, 4],相应的排序后的索引值为 [2, 1, 3, 0]。然后根据这个索引值,将 arr2 的元素同步调整,得到 [30, 20, 40, 10]。
原文地址: https://www.cveoy.top/t/topic/o3Wu 著作权归作者所有。请勿转载和采集!