NumPy索引和切片怎么使用

NumPy中的索引和切片可以帮助我们访问和操作数组中的元素。下面是一些常见的索引和切片操作示例：

1. 索引一个元素：

import numpy as np  arr = np.array([1, 2, 3, 4, 5]) print(arr[2])  # 输出：3 

2. 切片一个数组：

import numpy as np  arr = np.array([1, 2, 3, 4, 5]) print(arr[1:4])  # 输出：[2 3 4] 

3. 多维数组的索引和切片：

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]] 

4. 使用布尔索引：

import numpy as np  arr = np.array([1, 2, 3, 4, 5]) mask = arr > 2 print(arr[mask])  # 输出：[3 4 5] 

这些是一些常见的NumPy索引和切片操作示例，希望能帮助到你。