python - 如何使用字符串列表过滤 pandas DataFrame

Question

这个问题已经有答案了:

Filter dataframe rows if value in column is in a set list of values [duplicate] (7 个回答)

已关闭 5 年前。

我正在尝试过滤 pandas DataFrame df通过仅保留包含 mylist = ['cat', 'mouse'] 中的几个字符串之一的行，在特定列 df['Title'] :

  df.head()

                     Title               Duration   ...       
    0             The Cat1 & Mouse2       33 min    ...          
    1             Legend of the cat       10 min    ...        
    2             Foo-Bar                 3 min     ...      
    3             Legend of Mousopia      5 min     ...          
    4             Cat + Mouse             7 min     ...    

看着类似的问题，我尝试过过滤df通过执行以下操作:

z = df['Title'].str.lower()

df = df[z.contains([x for x in mylist])]

期待df.head()看起来像:

                     Title               Duration   ...       
    0             The Cat1 & Mouse2       33 min    ...          
    1             Legend of the cat       10 min    ...                 
    4             Cat + Mouse             7 min     ...    

但是，我不断收到以下错误:

AttributeError: 'Series' object has no attribute 'contains'

我已更新conda和pandas但不断得到相同的结果。

conda version : 4.5.4
conda-build version : 3.8.0
python version : 3.6.5.final.0
pandas version : 0.23.0           py36h830ac7b_0

我错过了什么？

Answer 1

尝试使用df['Title'].str.contains(*my_list)。