Python中如何依据一个numpy数组的顺序对另一个numpy数组排序

方案1：Numpy原生实现（推荐，大数据量性能更好） #

核心是通过两次argsort获取arr2的顺序映射，直接对数组做向量化操作，无需遍历元素。

• 代码示例：

import numpy as np

arr1 = np.array([1,2,3,4,5])
arr2 = np.array([2,3,1,5,4])

# 生成arr2的顺序秩次
rank = np.argsort(np.argsort(arr2))
# 排序arr1后按秩次重排
res = np.sort(arr1)[rank]
print(res)

• 输出结果：
  [2 3 1 5 4]

方案2：Python原生字典映射实现（适合小数据量，逻辑更直观） #

通过字典存储arr2元素和对应位置的映射，自定义排序key实现重排。

• 代码示例：

import numpy as np

arr1 = np.array([1,2,3,4,5])
arr2 = np.array([2,3,1,5,4])

pos_map = {val: idx for idx, val in enumerate(arr2)}
res = np.array(sorted(arr1, key=lambda x: pos_map[x]))
print(res)

• 输出结果和方案1完全一致。

注意：两种方案都默认两个数组元素完全匹配无缺失，如果存在重复元素，优先选择方案1，结果更稳定。

内容的提问来源于stack exchange，提问作者Rauan Saturin