创建使用字典映射数据帧中相应键值的数据帧列

Question

希望创建一个数据框列，该列将从字典中获取数据并在 DF 中搜索该值。下面的例子：

DF1：

ColA    ColB    ColC    ColD
Dog      4.5     1.3     6.4
Cat      154      89      2
Frog     8        x       9

字典 = {'狗'：'ColC'，'猫'：'ColB'，'青蛙'：'ColD'}

所需的数据框：'NewCol' 列使用字典来查找正确的值

ColA    ColB    ColC    ColD    NewCol
Dog      4.5     1.3     6.4      1.3
Cat      154      89      2       154 
Frog     8        x       9        9

谢谢您的帮助！！

Answer 1

您可以使用查找：

df['new_col'] = df.set_index('ColA').lookup(Dictionary.keys(), Dictionary.values())

输出：

   ColA   ColB ColC  ColD new_col
0   Dog    4.5  1.3   6.4     1.3
1   Cat  154.0   89   2.0     154
2  Frog    8.0    x   9.0       9