numpy.ndarray 二维数组的截取
时间: 2023-10-23 13:14:19 浏览: 152
python NumPy ndarray二维数组 按照行列求平均实例
要截取 numpy.ndarray 二维数组中的一部分,可以使用切片(slice)操作来获取所需的子数组。切片操作可以通过指定起始索引和结束索引来截取数组的一部分。
例如,假设我们有一个名为 `arr` 的二维数组,我们可以使用以下代码截取其中的一部分:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 截取第二行及其后面的所有行,第一列及其后面的所有列
sub_arr = arr[1:, :]
print(sub_arr)
```
输出结果为:
```
[[4 5 6]
[7 8 9]]
```
在这个例子中,我们使用 `arr[1:, :]` 获取从第二行开始的所有行,并使用 `:` 表示获取所有列。这样,我们获取的子数组就是原数组的第二行及其后面的所有行,第一列及其后面的所有列。
阅读全文