将一个四维数组按照第一维分割
时间: 2024-05-11 07:13:51 浏览: 137
数组分割1
假设四维数组为arr,要按照第一维进行分割,可以使用numpy库中的split函数:
```python
import numpy as np
# 创建一个4x3x2x5的四维数组
arr = np.ones((4, 3, 2, 5))
# 按照第一维进行分割,分成两个数组
arr1, arr2 = np.split(arr, 2, axis=0)
# 打印分割后的数组形状
print(arr1.shape) # 输出 (2, 3, 2, 5)
print(arr2.shape) # 输出 (2, 3, 2, 5)
```
上述代码中,使用numpy库中的ones函数创建一个4x3x2x5的四维数组,然后使用split函数按照第一维进行分割,分成两个数组arr1和arr2。最后打印分割后的数组形状。
阅读全文