python - 将一列中的文本替换为另一列中的字典

Question

我的一列中有文本，另一列有相应的字典。我已经对文本进行了标记，并希望替换那些在相应字典中找到与键匹配的标记。文本和字典特定于 pandas 数据帧的每个记录。

import pandas as pd

data =[['1','i love mangoes',{'love':'hate'}],['2', 'its been a long time we have not met',{'met':'meet'}],['3','i got a call from one of our friends',{'call':'phone call','one':'couple of'}]]

df = pd.DataFrame(data, columns = ['id', 'text','dictionary']) 

输出的最终数据帧应该是

data =[['1','i hate mangoes'],['2', 'its been a long time we have not meet'],['3','i got a phone call from couple of of our friends']
df = pd.DataFrame(data, columns =['id, 'modified_text'])

我在 Windows 机器上使用 Python 3

Answer 1

您可以在压缩 2 列并分割句子后使用 dict.get 方法:

df['modified_text']=([' '.join([b.get(i,i) for i in a.split()]) 
                  for a,b in zip(df['text'],df['dictionary'])])
print(df)

<小时/>

输出:

id                                  text  \
0  1                        i love mangoes   
1  2  its been a long time we have not met   
2  3  i got a call from one of our friends   

                                   dictionary  \
0                            {'love': 'hate'}   
1                             {'met': 'meet'}   
2  {'call': 'phone call', 'one': 'couple of'}   

                                      modified_text  
0                                    i hate mangoes  
1             its been a long time we have not meet  
2  i got a phone call from couple of of our friends