python - Numpy：乘以两个排列

Question

我有两个排列表示为 Numpy 数组：

A：

     0  1  2  3
A = [1, 3, 2, 0]

和乙：

     0  1  2  3
B = [0, 2, 1, 3]

我怎样才能得到C = A * B，其中

0->1 * 1->2 = 2
1->3 * 3->3 = 3
2->2 * 2->1 = 1
3->0 * 0->0 = 0

C = (2, 3, 1, 0)

有效地使用 Numpy？

Answer 1

很简单，b[a]会做的伎俩：

>>> import numpy as np
>>> a = np.array([1, 3, 2, 0])
>>> b = np.array([0, 2, 1, 3])
>>> c = b[a]
>>> print c
[2 3 1 0]