python - 按特定列的可能前缀列表过滤数据框

Question

我想做的是:

options = ['abc', 'def']
df[any(df['a'].str.startswith(start) for start in options)]

我想应用一个过滤器，所以我只有“a”列中的值以给定选项之一开头的条目。

下一个代码可以工作，但我需要它使用多个前缀选项...

start = 'abc'
df[df['a'].str.startswith(start)]

错误信息是

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

阅读Truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all()但还没有了解如何这样做。

Answer 1

您可以将选项元组传递给 startswith

df = pd.DataFrame({'a': ['abcd', 'def5', 'xabc', '5abc1', '9def', 'defabcb']})
options = ['abc', 'def']
df[df.a.str.startswith(tuple(options))]

你得到

    a
0   abcd
1   def5
5   defabcb