python之将 3D 数组切成多个 2D 数组

我有一个形状为 (64,64,3) 的 numpy 数组。我们如何获得三个维度为 (64,64) 的数组。

请您参考如下方法:

您可以使用moveaxis 将分割轴一直向左移动，然后使用序列解包:

x,y,z = np.moveaxis(arr, -1, 0)