python - 有没有一种使用交错 'mosaic' 模式将 4 个小数组组合成一个大数组的有效方法？

Question

我有 4 个相同形状的方形阵列

array1 = 1*np.ones((10,10))
array2 = 2*np.ones((10,10))
array3 = 3*np.ones((10,10))
array4 = 4*np.ones((10,10))

我想将它们以交错的马赛克图案重新组合成一个大阵列，如下所示:

result = np.asarray([[1,2,1,2,...,1,2],\
                     [3,4,3,4,...,3,4],\
                     [1,2,1,2,...,1,2],\
                     ...
                     [3,4,3,4,...,3,4]])

哪里 result在两个维度上都是原始单个图像的两倍。

有没有一种有效的方法来做到这一点？

为了说明我的问题，我使用了包含常量值的数组，但 现实中 ，这 4 个数组将是不同的图像。

Answer 1

在 numpy 中交错数据的两种常见方法是:

A) 将每个源分配给空白结果数组的一个切片，对应于数据应该去的地方:

result = np.zeros((20, 20)) # allocate space
result[::2, ::2] = array1 # put those values in the appropriate spots
result[::2, 1::2] = array2
result[1::2, ::2] = array3
result[1::2, 1::2] = array4

B) 使用堆叠将数据放在一个数组中，然后重新整形以将数据以隔行扫描的方式展平。这通常需要一些反复试验，但在玩了 REPL 之后，我想出了:

result = np.hstack((np.dstack((array1, array2)), np.dstack((array3, array4)))).reshape(20, 20)