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