python - 如何根据短语存在创建新列？

Question

我想根据短语存在创建新列

这是我的数据

No   Body
1    Office software is already paid
2    Excel software is not paid yet
3    Power point software is already paid

我想根据某个短语的存在进行分类，这是我的代码，

countries1 = df.body.str.extract('(software|is already paid)', expand = False)
dummies1 = pd.get_dummies(countries1)
df_1 = pd.concat([df,dummies1],axis = 1)

结果是

No   Body                                   software   is already paid    
1    Office software is already paid        0          1
2    Excel software is not paid yet         1          0
3    Power point software is already paid   0          1

我期望的是

No   Body                                   software   is already paid    
1    Office software is already paid        1          1
2    Excel software is not paid yet         1          0
3    Power point software is already paid   1          1

我的代码有什么问题？或者我没有使用正确的功能

Answer 1

让我们尝试使用 extractall:

df.assign(**df.Body.str.extractall('(software|is already paid)')[0]
              .str.get_dummies().sum(level=0))

输出:

   No                                  Body  is already paid  software
0   1       Office software is already paid                1         1
1   2        Excel software is not paid yet                0         1
2   3  Power point software is already paid                1         1