python - 将文本中的值替换为 python 字典中的整数值并求其总和

Question

我有一个类似这样的数据框:

column1 column2
   1     apple,apple,apple
   2     ball,ball,ball
   3     cat,dog,eel
   4     dog,dog,dog
   5     apple,cat,eel
   6     apple,ball,cat

我有一本字典，每个单词都有值(value):

{apple:1,
 ball:2,
 cat:3,
 dog:4,
 eel:5}

我想使用这个字典来替换数据框中的值并找到每行的总和。我怎样才能做到这一点？

最后我想要这样的东西:

column1      column2          column3 
   1     apple,apple,apple      3
   2     ball,ball,ball         6 
   3     cat,dog,eel            12
   4     dog,dog,dog            12
   5     apple,cat,eel          9
   6     apple,ball,cat         6

Answer 1

IIUC split + explode 然后map值

df.column2.str.split(',').explode().map(d).sum(level=0)
Out[286]: 
0     3
1     6
2    12
3    12
4     9
5     6
Name: column2, dtype: int64
df['column3']=df.column2.str.split(',').explode().map(d).sum(level=0)