当我尝试从 python 中的 Pandas 数据框创建新列时，部分关键字匹配不起作用？

Question

我有一个数据框描述，如下所述

  Description

我正在尝试对描述列进行关键字搜索，并且我将关键字列表作为列表。

我当前的代码只检查完全匹配而不是部分匹配。如果行中存在多个关键字，这些关键字将被分隔符分隔并填充新列。

我的代码

data=pd.read_excel('path_to_datafile.xlsx')
keywords=['dinner','government','Agents','entertainment','Agent']
keywords_lower = [item.lower() for item in keywords]
s=set(keywords_lower)
data['Keyword'] = data['Description'].apply(lambda x: '/'.join(set(x.lower().split()).intersection(s)))

如何才能做到这一点？

Answer 1

extractall将完成这项工作，但您必须首先构建模式：

...
keywords_lower = [item.lower() for item in keywords]
pattern = '(' + '|'.join('(?:' + i + ')' for i in keywords_lower) + ')'
df['Keyword'] = df['Description'].str.extractall(pattern, re.I).groupby(level=0).agg('/'.join)

你会得到：

                       Description           Keyword
0  Government entertainment people  Govern/entertain
1                  Dinner with CFO            Dinner
2  Commission to Agents government      Agent/govern

（ pattern在这里'((?:dinner)|(?:govern)|(?:agent)|(?:entertain))' ）