使用 numpy 重塑二维图像数组答案

【问题标题】：Reshaping a 2D array of images using numpy使用 numpy 重塑二维图像数组
【发布时间】：2022-01-13 19:24:03
【问题描述】：

我有一个形状为 (10, 10, 1024, 1024, 3) 的 numpy 数组。这表示一个 10x10 的图像网格，每个形状为(1024, 1024, 3)（1024x1024 彩色图像）。我想将其重塑为一个形状为(10*1024, 10*1024, 3) 的数组，其中新图像左上角的 1024x1024 像素块对应于我原始数组的[0, 0] 索引。使用 numpy 执行此操作的最佳方法是什么？

【问题讨论】：

嗨，在寻找“最佳”方式之前，您尝试过什么？
我试过arr.reshape(10*1024, 10*1024, 3)，但由于输出图像混乱，订单似乎不对。
我想resize 和swapaxis 的某种组合会得到想要的结果，但我对这些功能还没有很好的直觉。

标签： python arrays numpy

【解决方案1】：

这应该可以完成工作：np.swapaxes(arr,1,2).reshape(10*1024, 10*1024, 3)。请注意，swapaxis 会生成一个形状为 (10, 1024, 10, 1024, 3) 的数组。

【讨论】：