我想找到包含字符串的行,就像这样:
DF[DF.col.str.contains("foo")]
然而,这失败了,因为有些元素是NaN:
ValueError:不能索引包含NA / NaN值的向量
所以我诉诸于混淆
DF[DF.col.notnull()][DF.col.dropna().str.contains("foo")]
有没有更好的办法?
我想找到包含字符串的行,就像这样:
DF[DF.col.str.contains("foo")]
然而,这失败了,因为有些元素是NaN:
ValueError:不能索引包含NA / NaN值的向量
所以我诉诸于混淆
DF[DF.col.notnull()][DF.col.dropna().str.contains("foo")]
有没有更好的办法?
当前回答
也可以使用query方法查询DataFrame的列,使用布尔表达式,如下所示:
df.query("a.str.contains('foo', na=False)")
或者更有趣的,我认为更容易读懂:
df.query("a.str.contains('foo')==True")
注意,您可能不会得到性能上的改进,但是可读性更强(可以说)。
其他回答
您还可以制模:
DF[DF.col.str.contains(pat = '(foo)', regex = True) ]
也可以使用query方法查询DataFrame的列,使用布尔表达式,如下所示:
df.query("a.str.contains('foo', na=False)")
或者更有趣的,我认为更容易读懂:
df.query("a.str.contains('foo')==True")
注意,您可能不会得到性能上的改进,但是可读性更强(可以说)。
我不知道为什么(实际上来这里是为了寻找答案),但这也是可行的,并且不需要替换所有nan值。
import pandas as pd
import numpy as np
df = pd.DataFrame([["foo1"], ["foo2"], ["bar"], [np.nan]], columns=['a'])
newdf = df.loc[df['a'].str.contains('foo') == True]
使用或不使用.loc都有效。
我不知道为什么这是可行的,正如我所理解的,当你用括号索引时,熊猫评估括号内的任何东西为真或假。我不知道为什么括号内的短语“额外布尔”有任何影响。
df[df.col.str.contains("foo").fillna(False)]
除了上面的答案,我想说的是,对于没有单一单词名称的列,你可以使用:-
df[df['Product ID'].str.contains("foo") == True]
希望这能有所帮助。