savetxt用法 如何将三维数组结果保存到Python/numpy的4D数组中?



python多维数组 (1)

您需要在连接之前添加一个轴,例如

import numpy as np
arrs = [np.random.random((32, 32, 3))
        for i in range(50)]

res = np.concatenate([arr[np.newaxis] for arr in arrs])
res.shape
# (50, 32, 32, 3)

编辑:或者,在这种情况下,你可以简单地在你的数组列表中调用np.array

res = np.array(arrs)
res.shape
# (50, 32, 32, 3)

我正在阅读有关32 X 32 RGB图像的信息。 所以它是一个形状为(32,32,3)的三维数组,它保存了R,G和B的颜色

现在,我想读取50个这样的图像,并制作这些图像的数组。 所以我决定做一个四维数组,它的尺寸是(50,32,32,3),这里的第一维是50,图像的数量是第二维,第三维和第四维是图像的尺寸是(32, 32,3)

我试图使用连接,但我得到错误。 有没有办法做到这一点?





multidimensional-array