如何获取包含特定单词的行？

Question

我正在使用netfilx dataset 。由Victor Soeiro上传到Kaggle。我想根据流派制作新数据集。例如，如果类型包含Word ['Comedy']或['Comedy'，'Fantasy']，我想要新的数据框。

my_code

df_comedy = df[df['genres'].str.extract("'comedy'")]

它给了我这个错误。 valueerror：模式不包含捕获组

如果您不了解我的问题，请我提供更多信息。

谢谢。

Answer 1

这是解决您的问题的解决方案：

import pandas as pd
#reading csv file
df = pd.read_csv("Assignment.csv")

# filtering the rows where Credit-Rating is Fair
df = df[df['Credit-Rating'].str.contains('Fair')]
print(df)

输出：

job        Age_Range    Salary    Credit-Rating  Savings    

Own        Middle-aged  High       Fair          10000    
Govt       Young        Low        Fair          15000   
Private    Senior       Average    Fair          20000  
Own        Middle-aged  High       Fair          13000   
Own        oung         Low        Fair          17000  

使用包含而不是提取，您将获得答案。