NumPy索引和切片怎么使用
扬州沐宇科技
2024-05-13 11:22:20
numpy
NumPy中的索引和切片可以帮助我们访问和操作数组中的元素。下面是一些常见的索引和切片操作示例：
- 索引一个元素：
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr[2]) # 输出：3
- 切片一个数组：
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr[1:4]) # 输出：[2 3 4]
- 多维数组的索引和切片：
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[1, 2]) # 输出：6
print(arr[:, 1]) # 输出：[2 5 8]
print(arr[1:, :2]) # 输出：[[4 5] [7 8]]
- 使用布尔索引：
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
mask = arr > 2
print(arr[mask]) # 输出：[3 4 5]
这些是一些常见的NumPy索引和切片操作示例，希望能帮助到你。
下一条：如何选择java xml的解析库
相关问答
