这里有很多关于排序数组的问题，但我找不到解决我的问题的问题（这让我希望我没有想得太复杂）。所以我们开始：
我有两个值相等但顺序不同的数组：
a = np.array([4, 2, 5, 6])
# b = np.random.permutation(a) # general case
b = np.array([5, 2, 6, 4])
我需要将 b 排序为 a 的索引，所以在这里：
ind = np.array([3, 1, 0, 2])
请注意，我不想更改 a。
【问题讨论】：
标签： python arrays sorting numpy
broadcasting & np.where 的一种方法-
_,C = np.where(a[:,None] == b)
示例运行 -
In [210]: a = np.array([4, 2, 5, 6])
In [211]: b = np.array([5, 2, 6, 4])
In [212]: _,C = np.where(a[:,None] == b)
In [213]: C
Out[213]: array([3, 1, 0, 2], dtype=int64)
【讨论】：
a 和 b 很大，这将构建一个 len(a) x len(b) 的布尔数组，根据用例的不同，该数组可能过多。
在这种情况下，a 已经排序。因此，您只需要b 的排序索引。
但是，如果 a 没有排序，你可以这样做
a = array([2, 4, 5, 6])
b = array([5, 2, 6, 4])
b.argsort()[a.argsort()] # array([1, 3, 0, 2])
这使用argsort 方法返回排序b 和a 所需的索引。
另一个例子：
a = array([4, 2, 5, 6])
b = array([5, 2, 6, 4])
b.argsort()[a.argsort()] # array([3, 1, 0, 2])
【讨论】：
a = np.array([4, 4, 2, 5, 6]) 和 b = np.array([5, 2, 4, 6, 4])。
argsort。
b = a[a.argsort()[b.argsort().argsort()]]（但更正了放错位置的括号）。
np.argsort 是它自己的逆。如果y = x[x.argsort()]，那么x = y[x.argsort().argsort()]。具体来说，您使用a.argsort() 将a 转换为排序数组，并使用b.argsort().argsort() 从排序数组转换为b。因此，您可以写：
b = a[a.argsort()][b.argsort().argsort()]
换句话说：
ind = a.argsort()[b.argsort().argsort()]
【讨论】：
b = a[a.argsort()[b.argsort().argsort()]]（注意右括号），从第二个 sn-p 可以清楚地看出。