如何进行关键字搜索并将所选列中存在的关键字列出到熊猫数据框的新列？

Question

我有一个包含 foodhist 列的 df

它可以包含'cheeseburgers,largepizzafrompizzahut'作为输入

df['foodhist']

我有一个列表unhealthyfoods = ['burger','pizza' etc.]

我想添加 df['unhealthyfoodyouhad'] 列将在df['foodhist'] 中查找来自unhealthyfoods 的所有项目并将它们添加到此列。

df['foodhist'] 中的数据不能间隔，不能分隔，但我还是想寻找关键字的出现并在df['unhealthyfoodyouhad'] 中返回它们。

为此，在 'cheeseburgers,largepizzafrompizzahut' 中应在 df['unhealthyfoodyouhad'] 列中添加 burger,pizza。

Answer 1

试试这个：

import re

unhealthyfoods=['burger','pizza']

df=pd.DataFrame({'foodhist':['cheeseburger','burgerandlargepizza']})
list1=[]
for food in unhealthyfoods:
    list1.append([re.findall(food,x) for x in df['foodhist']])
df['unhealthyfoodyouhad']=pd.DataFrame(list1).T.sum(1)
df['unhealthyfoodyouhad']=[','.join(i) for i in df['unhealthyfoodyouhad']]

输出：

df
Out[95]: 
              foodhist unhealthyfoodyouhad
0         cheeseburger              burger
1  burgerandlargepizza        burger,pizza