使用 numpy 重塑二维图像数组

Question

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

Answer 1

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