numpy数组切片对切片
时间: 2024-05-11 17:10:22 浏览: 118
numpy数组切片操作可以通过冒号分隔切片参数来进行。下面是一些常见的numpy数组切片操作示例:
1. 一维数组切片:
- 一个参数:`a[i]`,表示取数组中索引为`i`的元素。
- 两个参数:`b = a[i:j]`,表示取数组中索引从`i`到`j-1`的元素。
- 三个参数:`b = a[i:j:s]`,表示以步长`s`取数组中索引从`i`到`j-1`的元素。
2. 二维数组切片:
- 取元素:`X[n0, n1]`,表示取二维数组`X`中第`n0`行、第`n1`列的元素。
- 切片:`X[s0:e0, s1:e1]`,表示取二维数组`X`中行索引从`s0`到`e0-1`,列索引从`s1`到`e1-1`的元素。
- 特殊情况:`X[:e0, s1:]`,表示取二维数组`X`中前`e0`行,从`s1`列到最后一列的元素。
下面是一些代码示例:
```python
import numpy as np
# 一维数组切片
a = np.array([1, 2, 3, 4, 5])
print(a[2]) # 输出:3
print(a[1:4]) # 输出:[2 3 4]
print(a[::2]) # 输出:[1 3 5]
# 二维数组切片
X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(X[1, 2]) # 输出:6
print(X[0:2, 1:3]) # 输出:[[2 3] [5 6]]
print(X[:2, 1:]) # 输出:[[2 3] [5 6]]
```
阅读全文