在Python中创建包含某些单词的文档列表[重复]

Question

我有一个 CSV 文件，其中包含以下列：

df =
title body
A     yellow, gree
B     blue, red
C     red, green
D     grey, blue

我想创建一个包含红色单词的行的新数据框：

df_New =
title body
B     blue, red
C     red, green

到目前为止，我的代码如下所示：

import pandas

Documents = pandas.read_csv(df.csv)

words = ['red', 'Red']

Df_New = Documents.str.match('(words)')

Answer 1

您可以使用.str.contains()

df_ = df[df['body'].str.contains('red', case=False)]

print(df_)

  title        body
1     B   blue, red
2     C  red, green