numpy二维数组的切片
时间: 2024-01-06 13:25:24 浏览: 171
以下是对NumPy二维数组切片的介绍和示例代码:
NumPy中的二维数组切片可以通过指定行和列的范围来获取数组的子集。切片的格式为`array[start_row:end_row, start_col:end_col]`,其中`start_row`和`end_row`表示行的起始和结束索引,`start_col`和`end_col`表示列的起始和结束索引。
示例代码如下:
```python
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取整个数组
slice1 = arr[:, :]
print(slice1)
# 输出:
# [[1 2 3]
# [4 5 6]
# [7 8 9]]
# 获取第一行
slice2 = arr[0, :]
print(slice2)
# 输出:[1 2 3]
# 获取第一列
slice3 = arr[:, 0]
print(slice3)
# 输出:[1 4 7]
# 获取子矩阵
slice4 = arr[1:3, 1:3]
print(slice4)
# 输出:
# [[5 6]
# [8 9]]
```
阅读全文