python - 使用字典映射 numpy 数组？

Question

我有一个可能很长的 1 和 0 数组，如下所示:a = [0,0,1,0,1,0,.......]我想将每对连续的值转换为 0 和 3 之间的整数，如下所示:

• 0,0 -> 0
• 0,1 -> 1
• 1,0 -> 3
• 1,1 -> 2

我正在寻找一种干净高效的方法来使用上面的映射创建数组 b(下面的示例输出):b = [0, 3, 3,...]

字典是个好主意吗？不过我不知道该怎么做。

Answer 1

尝试:

x = np.reshape(a, (-1,2))

b = x[:,0]*2 + (x[:,0] ^ x[:,1])