在这个多维数组中构造数据

时间:2018-10-20 08:53:35

标签: python numpy multidimensional-array

我正在用python处理图像,出于特定目的,我被要求 “制作具有以下尺寸的numpy数组:32x3x512x512(补丁x颜色通道,高度,宽度)。”我获取任何数据都没有问题,但是我无法描绘numpy数组的结构。我最好的方法是: Option 1Option 2。而且我可能还没有接近。

关于numpy数组的外观有更好的主意吗?

1 个答案:

答案 0 :(得分:1)

您可以从字面上理解问题的描述并将其用作数组的形状:

arr = np.empty((32, 3, 512, 512), 'u1')

现在arr[0]是第一个色块(即第一张图像),形状为(3, 512, 512),因此每个像素为24(3x8)位,尺寸为512 x 512像素。

某些人可能会使用不同的维度顺序,例如(32, 512, 512, 3),但这实际上取决于您最终要执行的处理。