python - 有没有办法将 numpy.ndarray 的每个条目包装到一个单独的数组中？

Question

我在将数组调整为正确的形状以将其用作卷积神经网络的输入时遇到一些问题:

我的数组的形状为(100,64,64)，但我需要它为(100,64,64,1)。我意识到这看起来有点奇怪，但我基本上想将每个条目打包到一个单独的数组中。

一个简化的示例，使用 2D 数组，其中类似的内容是从 (3,3) 到 (3,3,1):

[[0,1,0],        [[[0],[1],[0]],
 [1,1,1],         [[1],[1],[1]],
 [0,0,1]]         [[0],[0],[1]]]

有没有一种方便的方法使用 numpy 来做到这一点？

我尝试使用函数numpy.reshape:我知道如何“添加”另一个数组来包装原始数组。

import numpy as np

data = data.reshape((1,)+data.shape)

这给出了data.shape的输出:(1,100,64,64)。有没有办法在“内端”添加尺寸？

如果我尝试 data.reshape(data.shape+(,1))，我会收到无效语法错误。

Answer 1

作为docs points out ，当形状与您的形状兼容时，您也可以直接更改数组的形状:

a = np.array([
    [0, 1, 0],
    [1, 1, 1],
    [0, 0, 1]
])

a.shape += (1,)
a

# array([[[0], [1], [0]],
#        [[1], [1], [1]],
#        [[0], [0], [1]]])