pandas - 根据 pandas 中的模式匹配创建列

Question

我有一个包含“名称”和“任务”两列的日期框架。我想根据列表中的匹配条件创建名为“task_category”的第三列。请注意，以下数据仅作为示例，我实际上有 100 种模式要查找，而不是下面显示的三种。

df = pd.DataFrame(
        {'Name': ["a","b","c"],
        'Task': ['went to trip','Mall Visit','Cinema']})
task_category=['trip','Mall','Cinema']

    Name    Task        task_category
0   a   went to trip    trip
1   b   Mall Visit      Mall
2   c   Cinema          Cinema

Answer 1

使用 series.str.extract()

pat=r'({})'.format('|'.join(task_category))
#'(trip|Mall|Cinema)'
df['task_category']=df.Task.str.extract(pat)
print(df)

---

  Name          Task task_category
0    a  went to trip          trip
1    b    Mall Visit          Mall
2    c        Cinema        Cinema