Python Pandas:String包含但不包含

2024-10-04 15:28:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图匹配Pandas数据帧中包含和不包含特定字符串的行。例如:

import pandas
df = pandas.Series(['ab1', 'ab2', 'b2', 'c3'])
df[df.str.contains("b")]

输出:

0    ab1
1    ab2
2     b2
dtype: object

期望输出:

2     b2
dtype: object

问:有没有一种优雅的表达方式?

df[[df.str.contains("b")==True] and [df.str.contains("a")==False]]
# Doesn't give desired outcome

Tags: 数据字符串importpandasdfobjectb2series
3条回答

您可以使用.loc和~索引:

df.loc[(df.str.contains("b")) & (~df.str.contains("a"))]

2    b2
dtype: object

你快到了,你只是语法不太正确,应该是:

df[(df.str.contains("b") == True) & (df.str.contains("a") == False)]

如果有很多条件要应用,另一种可能更干净的方法是将过滤器与reduce或循环链接在一起:

from functools import reduce
filters = [("a", False), ("b", True)]
reduce(lambda df, f: df[df.str.contains(f[0]) == f[1]], filters, df)
#outputs b2

或者:

>>> ts.str.contains('b') & ~ts.str.contains('a')
0    False
1    False
2     True
3    False
dtype: bool

或者使用regex:

>>> ts.str.contains('^[^a]*b[^a]*$')
0    False
1    False
2     True
3    False
dtype: bool

相关问题 更多 >

    热门问题