是否有一种优雅的方法将numpy数组拆分为多个框？

Question

我有一个当前位于numpy数组中的图像。这对我的工作非常有用，因为我可以对图像进行傅立叶变换。

目前，我有兴趣查看图像很多位的傅立叶变换。该图像是numpy中的512x512数组，我想将其拆分为小盒子。我正在考虑将其分成4盒然后是16盒。

我已经找到np.array_split，但是这似乎只能将数组拆分为条带。结果，我多次调用它，如下所示（其中Real是512,512的输入数组）

H1, H2 = np.array_split(Real, 2, axis=0)
B1, B2 = np.array_split(H1, 2, axis=1)
B3, B4 = np.array_split(H2, 2, axis=1)

fig, ax = plt.subplots(2, 2)
ax[0,0].imshow(B1, cmap='Greys')
ax[0,1].imshow(B2, cmap='Greys')
ax[1,0].imshow(B3, cmap='Greys')
ax[1,1].imshow(B4, cmap='Greys')

plt.show()

当我将它们分成16个时，感觉好像会很不整洁。我想知道我是否没有注意到我可以使用的另一个numpy函数？

谢谢