python三维数组转多个二维数组

源码资源网

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

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

复制代码

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