python - 创建使用另一列中的字符串映射字典的数据框

Question

使用字典（下面的示例），需要创建一个新的数据框列，该列将示例数据框中的不同列，并且字符串中的第一个单词与字典键匹配，将值分配给新列。下面的例子：

dictionary = {'dog':'yellow', 'cat':'black, 'frog':'green', 'horse':'brown'}

原DF：

 ColA:
 The dog and horse ate food
 Where is the frog?
 horse and cat and frog walked together

所需的DF：

 ColA:                                      ColB
 The dog and horse ate food               yellow
 Where is the frog?                       green
 horse and cat and frog walked together   brown

有什么建议么？谢谢！

标签： pythonpandasdataframedictionary

Answer 1

尝试：

df["ColB"] = df["ColA"].str.split(expand=True).apply(lambda x: x.str.strip('.?,!":;').map(dictionary)).bfill(axis=1)[0]

                                     ColA    ColB
0              The dog and horse ate food  yellow
1                      Where is the frog?   green
2  horse and cat and frog walked together   brown