三维数组转多个二维数组

Question

现在有一个360* 640 4的三维数组，想把它转成4个360640的二位数组，各位能教一下吗，谢谢！

Answer 1

当你有一个360x640x4的三维数组时，你可以使用Python中的切片操作来将其转换为4个360x640的二维数组。以下是一个示例代码：

import numpy as np
 # 创建一个360x640x4的三维数组
three_dim_array = np.random.randint(0, 10, size=(360, 640, 4))
 # 将三维数组转换为四个二维数组
two_dim_arrays = np.split(three_dim_array, 4, axis=2)
 # 打印每个二维数组的形状
for i, arr in enumerate(two_dim_arrays):
    print(f"二维数组 {i+1} 的形状：{arr.shape}")

在上述代码中，我们使用NumPy库创建了一个随机的360x640x4的三维数组 three_dim_array 。然后，我们使用 np.split() 函数将该三维数组沿着第三个维度（axis=2）分割成四个二维数组。最后，我们打印了每个二维数组的形状。