在Python中重构数组的形状

Question

我在Python中有一个带有形状（64,64）的数组，我希望以我可以拥有形状（64,64,3）的数组的方式重复这些元素（三次）。有什么想法吗？

Answer 1

可能最简单的方法是使用numpy.dstack：

import numpy as np

b = np.dstack((a, a, a))

其中a是原始数组（形状64×64），b是新数组（形状64×64×3）。