python - python找到排列映射

Question

我想找到给定两个数组 A 和 B 的映射，它们是彼此的排列。一个等效的函数是

import numpy as np
def perm_map(A, B):
    return [np.where(np.array(B) == el)[0][0] for el in A]

print( perm_map(['A', 'B', 'C'], ['B', 'C', 'A']) )

此示例的结果是 [2, 0, 1]

问题：

• 我正确地调用它吗？
• 是否有内置的高效功能？

标签： pythonpermutation