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索引和切片操作示例，希望能帮助到你。