python - 在 Python 中创建一个 5D 数组

Question

我有一个灰色图像，我想在其中将每个像素映射到其他 N 个大小为 LxM 的矩阵。如何初始化这样的矩阵？我试过 result=numpy.zeros(shape=(i_size[0],i_size[1],N,L,M))了，但我得到了 Value Error 'array is too big'。任何人都可以建议一个替代方法？

Answer 1

N如果我理解正确，灰色图像中的每个像素都映射到其他图像中的单个像素。在这种情况下，地图数组是numpy.zeros((i.shape[0], i.shape[1], N, 2), dtype=numpy.int32)因为您需要将 1 x 和 1 y 坐标存储到彼此N的数组中，而不是每次都存储完整的第 N 个数组。使用整数索引将进一步减少内存使用。

然后result[y,x,N,0]和result[y,x,N,1]是y和x到第 N 个图像的映射。