在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）。