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

时间:2020-09-18 13:10:44

标签: python pandas dataframe

试图创建一个数据框列,该列将从字典中获取数据并在DF中搜索该值。下面的示例:

DF1:

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

Dictionary = {'Dog':'ColC','Cat':'ColB','Frog':'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

感谢您的帮助!

1 个答案:

答案 0 :(得分:5)

您可以使用查找:

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