通过包含str过滤熊猫数据框行

Question

我有df很多行的python pandas数据框。从这些行中，我想切出并且仅使用“ body”列中包含单词“ ball”的行。为此，我可以这样做：

df[df['body'].str.contains('ball')]

问题是，我希望它不区分大小写，这意味着如果出现Ball或bAll一词，我也希望它们。进行不区分大小写的搜索的一种方法是将字符串转换为小写，然后以这种方式搜索。我想知道如何去做。我试过了

df[df['body'].str.lower().contains('ball')]

但这是行不通的。我不确定是否应该在此等性质上使用lambda函数。

Answer 1

您可以.str再次使用来访问字符串方法，或者（更好的是IMHO）用于case=False保证不区分大小写：

>>> df = pd.DataFrame({"body": ["ball", "red BALL", "round sphere"]})
>>> df[df["body"].str.contains("ball")]
   body
0  ball
>>> df[df["body"].str.lower().str.contains("ball")]
       body
0      ball
1  red BALL
>>> df[df["body"].str.contains("ball", case=False)]
       body
0      ball
1  red BALL
>>> df[df["body"].str.contains("ball", case=True)]
   body
0  ball

（请注意，如果您要进行分配，则最好使用df.loc，以避免可怕的SettingWithCopyWarning，但是如果我们只是在此处选择，则无所谓。）

（注2：猜想我真的不需要在那里指定“回合”。）