jupyter数组分割
时间: 2024-09-25 17:11:55 浏览: 39
python numpy–数组的组合和分割实例
在Jupyter Notebook中,如果你正在处理NumPy数组并希望对其进行分割,你可以使用`numpy.split()`函数。这个函数可以根据指定的分隔点、大小或区间将数组分割成几个子数组。下面是一个基本的例子:
```python
import numpy as np
# 创建一个数组
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
# 按索引分割
split_at_index = np.split(arr, [3, 7]) # 分割点分别为3和7
print(split_at_index)
# 或者按等份分割
split_by_size = np.split(arr, len(arr), axis=0) // 轴0表示沿行方向
print(split_by_size)
```
在这个例子中,`np.split(arr, [3, 7])`会返回两个子数组,第一个包含前四个元素 `[0, 1, 2, 3]`,第二个包含剩余的元素 `[4, 5, 6, 7, 8, 9]`。
`np.split(arr, len(arr), axis=0)` 则将整个数组平均分成两半,因为`len(arr)`就是数组的长度。
阅读全文