我想找到包含字符串的行,就像这样:

DF[DF.col.str.contains("foo")]

然而,这失败了,因为有些元素是NaN:

ValueError:不能索引包含NA / NaN值的向量

所以我诉诸于混淆

DF[DF.col.notnull()][DF.col.dropna().str.contains("foo")]

有没有更好的办法?


当前回答

我不知道为什么(实际上来这里是为了寻找答案),但这也是可行的,并且不需要替换所有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['Product ID'].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都有效。

我不知道为什么这是可行的,正如我所理解的,当你用括号索引时,熊猫评估括号内的任何东西为真或假。我不知道为什么括号内的短语“额外布尔”有任何影响。

也可以使用query方法查询DataFrame的列,使用布尔表达式,如下所示:

df.query("a.str.contains('foo', na=False)")

或者更有趣的,我认为更容易读懂:

df.query("a.str.contains('foo')==True")

注意,您可能不会得到性能上的改进,但是可读性更强(可以说)。

df[df.col.str.contains("foo").fillna(False)]

您还可以制模:

DF[DF.col.str.contains(pat = '(foo)', regex = True) ]