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]

问题：

• 我叫正确吗？
• 是否有内置的高效功能？