布尔索引是指在 numpy 数组中使用布尔值(True
或 False
)作为索引来选择数组中的元素。
在使用布尔索引时,首先需要创建一个布尔型数组,其中数组的每个元素都是 True
或 False
。然后,使用该数组作为索引即可选择数组中对应位置为 True
的元素。
例如,假设有一个 numpy 数组 a
,其中包含了一些整数:
import numpy as np
a = np.array([1, 2, 3, 4, 5])
现在,我们想要选择所有大于 2 的元素。可以通过以下步骤实现:
- 创建一个布尔型数组,其中数组的每个元素都是
True
或False
。可以使用比较运算符(例如>
、<
、==
等)来创建布尔型数组:mask = a > 2
- 使用布尔型数组
mask
作为索引来选择数组中对应位置为True
的元素:b = a[mask] # 选择大于 2 的元素
3、选择‘False’
b = a[~mask] # 选择小于 2 的元素