学科分类
目录
数据分析

布尔型索引的基本使用

布尔型索引指的是将一个布尔数组作为数组索引,返回的数据是布尔数组中True对应位置的值。

假设现在有一组存储了学生姓名的数组,以及一组存储了学生各科成绩的数组,存储学生成绩的数组中,每一行成绩对应的是一个学生的成绩。如果我们想筛选某个学生对应的成绩,可以通过比较运算符,先产生一个布尔型数组,然后利用布尔型数组作为索引,返回布尔值True对应位置的数据。示例代码如下:

In [72]: # 存储学生姓名的数组
         student_name = np.array(['Tom', 'Lily', 'Jack', 'Rose'])
In [73]: student_name
Out[73]: array(['Tom', 'Lily', 'Jack', 'Rose'], dtype='<U4')
In [74]: # 存储学生成绩的数组
         student_score = np.array([[79, 88, 80], [89, 90, 92], 
                                   [83, 78, 85], [78, 76, 80]])
In [75]: student_score
Out[75]: 
array([[79, 88, 80],
​     [89, 90, 92],
​     [83, 78, 85],
​     [78, 76, 80]])
In [76]: # 对student_name和字符串“Jack”通过运算符产生一个布尔型数组
         student_name == 'Jack'
Out[76]: array([False, False, True, False])
In [77]: # 将布尔数组作为索引应用于存储成绩的数组student_score,
         # 返回的数据是True值对应的行
         student_score[student_name=='Jack']
Out[77]: array([[83, 78, 85]])

布尔索引的相关示意图如图1所示。

img

图1 布尔索引图示

需要注意的是,布尔型数组的长度必须和被索引的轴长度一致。

此外,我们还可以将布尔型数组跟切片混合使用,示例代码如下:

In [78]: student_score[student_name=='Jack', :1]
Out[78]: array([[83]])

值得一提的是,使用布尔型索引获取值的时候,除了可以使用“==”运算符,还可以使用诸如“!=”、“-”来进行否定,也可以使用“&”、“|”等符号来组合多个布尔条件。

点击此处
隐藏目录