如何将np.array_split()函数返回的一个数组列表拆分成两个数组
时间: 2024-05-06 08:20:56 浏览: 133
详谈Numpy中数组重塑、合并与拆分方法
可以通过索引将返回的数组列表拆分成两个数组,如下所示:
```python
import numpy as np
arr = np.arange(10)
sub_arrays = np.array_split(arr, 2)
# 拆分为两个数组
arr1 = sub_arrays[0]
arr2 = sub_arrays[1]
```
在这个例子中,`np.array_split()`函数将长度为10的数组拆分成两个长度为5的数组,并将结果存储在`sub_arrays`列表中。我们可以通过`sub_arrays`列表的索引来获取拆分后的两个数组,并将它们分别存储在`arr1`和`arr2`中。
阅读全文