使用dataframe和list之间的重叠字数创建新的dataframe列

Question

我在修复以下问题时遇到问题： 我有一个数据框，每一行上都有标记文本，看起来（有些东西）如下

index feelings           
1     [happy, happy, sad] 
2     [neutral, sad, mad] 
3     [neutral, neutral, happy]

和单词列表lst1=[happy, fantastic]，lst2=[mad, sad]，lst3=[neutral]，我想检查我的数据帧中每一行的单词在列表中出现的次数。因此，输出将如下所示：

index feelings                  occlst1 occlst2 occlst3      
1     [happy, happy, sad]       2      1        0
2     [neutral, sad, mad]       0      2        1
3     [neutral, neutral, happy] 1      0        2

因此，我想创建一个新列，并将dataframe单元格与列表进行比较

提前谢谢

Answer 1

您还可以使用：

my_lists = [lst1, lst2, st3]
occ = pd.DataFrame.from_records(df['feelings'].apply(lambda x: [pd.Series(x).isin(l).sum() for l in my_lists]).values, columns=['occlst1', 'occlst2', 'occlst3'])
df_occ = df.join(occ)

输出：

                    feelings  occlst1  occlst2  occlst3
0        [happy, happy, sad]        2        1        0
1        [neutral, sad, mad]        0        2        1
2  [neutral, neutral, happy]        1        0        2