如何从Pandas Series获取基于字符串匹配的True/False掩码？

嘿，这个需求处理起来超简单的！Pandas本身就提供了非常直观的方式来完成这个操作，下面给你两种常用的方法：

方法一：直接使用相等运算符 == #

这是最直观的方式，直接把Series和目标字符串用==比较，就能得到由True/False组成的布尔序列：

import pandas as pd

# 定义你的Series
s = pd.Series(['a', 'b', 'c', 'd', 'a'], index=[1, 2, 3, 4, 5])
# 生成布尔序列，匹配所有等于'a'的位置
bool_mask = s == 'a'

# 输出结果
print(bool_mask)

运行后你会得到正好符合预期的结果：

1     True
2    False
3    False
4    False
5     True
dtype: bool

方法二：使用Series的eq()方法 #

如果你喜欢用Pandas的方法式调用，也可以用eq()方法来实现完全一样的效果，这种方式在链式操作中会更顺手：

bool_mask = s.eq('a')

执行后得到的布尔序列和上面的方法完全一致。

另外，如果之后你需要提取这些匹配到的元素，直接用这个布尔序列来索引原Series就可以了：s[bool_mask]，会返回所有值为'a'的元素。

内容的提问来源于stack exchange，提问作者prre72