|
|
python三维数组转多个二维数组
当你有一个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）分割成四个二维数组。最后，我们打印了每个二维数组的形状。
|
|
发表于 2024-2-26 10:11:58
扫码关注微信公众号，及时获取最新资源信息！【