Python 如何将形状（32,32）为（1,32,32）的numpy数组存储

出于各种原因，我需要将形状（32,32）的2d numpy数组转换为3d形状（1,32,32）。有办法做到这一点吗？

只需使用：

我使用了一组零来演示

>>> a = np.zeros((32, 32))
>>> a.reshape((1, 32, 32))
array([[[ 0.,  0.,  0., ...,  0.,  0.,  0.],
        [ 0.,  0.,  0., ...,  0.,  0.,  0.],
        [ 0.,  0.,  0., ...,  0.,  0.,  0.],
        ..., 
        [ 0.,  0.,  0., ...,  0.,  0.,  0.],
        [ 0.,  0.,  0., ...,  0.,  0.,  0.],
        [ 0.,  0.,  0., ...,  0.,  0.,  0.]]])
>>>