python - 给定重排 A 和 B，计算 P 使得 A[P] == B

Question

如果 A 和 B 是对应于相同(不同)元素的两个排序的两个数组，则存在唯一索引数组 P这样 A[P] 等于 B。例如，如果 A 和 B 是

A = ['b', 'c', 'e', 'd', 'a']
B = ['a', 'd', 'c', 'b', 'e']

那么所需的P是

P = [4, 3, 1, 0, 2]

numpy(或标准 Python)是否有计算这样一个 P 的函数？

Answer 1

使用标准的python

>>> A = ['b', 'c', 'e', 'd', 'a']
>>> B = ['a', 'd', 'c', 'b', 'e']
>>> P = [ A.index(i) for i in B ]
>>> P
[4, 3, 1, 0, 2]