You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何在Pandas中筛选字符串列值以列表任意元素开头的行?

筛选Pandas DataFrame中值以指定列表元素开头的行

嘿,这个需求太常见了!我平时处理数据经常碰到,给你几个实用的方法,都能轻松搞定你的问题:

方法一:用str.startswith(最推荐)

Pandas的str.startswith方法本身就支持传入多个前缀,只需要把你的列表转成元组就行(新版本Pandas其实也支持直接传列表,不过转元组兼容性更好)。代码非常简洁:

import pandas as pd

df = pd.DataFrame({'a': ['aa10', 'aa11', 'bb13', 'cc14']})
valids = ['aa', 'bb']

# 核心筛选代码
filtered_df = df[df['a'].str.startswith(tuple(valids))]

# 输出结果
print(filtered_df)

运行后会得到:

a
0  aa10
1  aa11
2  bb13

这个方法的优势是原生矢量化操作,速度快,代码可读性拉满,完全不需要额外处理正则那些复杂的东西。

方法二:正则表达式匹配

如果你需要更灵活的匹配规则(比如后续要加其他正则条件),可以用str.match来实现:

import pandas as pd
import re

df = pd.DataFrame({'a': ['aa10', 'aa11', 'bb13', 'cc14']})
valids = ['aa', 'bb']

# 构建前缀匹配的正则模式
pattern = '^(' + '|'.join(valids) + ')'
# 如果valids里有正则特殊字符(比如.、*),记得用re.escape转义:
# pattern = '^(' + '|'.join(re.escape(v) for v in valids) + ')'

filtered_df = df[df['a'].str.match(pattern)]

这里的^表示从字符串开头匹配,|是逻辑或,所以^(aa|bb)就会匹配所有以aa或bb开头的字符串。

方法三:apply遍历(不推荐大数据集)

如果你的数据量很小,也可以用apply结合lambda表达式实现,但这个方法是逐行遍历,大数据集下速度会很慢,所以只适合小数据场景:

filtered_df = df[df['a'].apply(lambda x: any(x.startswith(v) for v in valids))]

总结一下,优先用方法一,简洁高效;如果需要复杂匹配就用方法二;方法三尽量少用除非数据量极小。

内容的提问来源于stack exchange,提问作者Hanshan

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.26 09:15:33