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

Question

我正在使用 netfilx 数据集。 由 VICTOR SOEIRO 在 KAGGLE 上传。 我想根据流派制作新的数据集。 例如，如果流派包含单词 ['comedy'] 或 ['comedy','fantasy']，我想要新的数据框。

我的代码

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  

使用contains而不是extract ，你会得到你的答案。