熊猫-如果存在列条目,则从数据框中映射值

时间:2019-06-11 20:56:48

标签: python pandas

我有2个数据帧df1df2df2df1短。如果pred中存在df2中的df1,我试图将元素pedestriandf2复制到df1。我看过很多帖子,但找不到解决我的问题的帖子。

使用地图会在NaN的任何地方向我返回pred

d = df2.set_index('pedestrian')['pred'].to_dict()
df1['pred'] = df2['pred'].map(d)

如果使用replace,程序不会停止运行。

d = df2.set_index('pedestrian')['pred'].to_dict()
df1.replace(d)

下面是df1, df2和我想要的输出。

df1

   pred   true            pedestrian     
0   -1     1    /data/jaad/image/0316/pedestrian/00...
1   -1     0    /data/jaad/image/0316/ped1/0001.png 
2   -1     0    /data/jaad/image/0316/ped2/0001.png

df2

      pred    true            pedestrian
0   0.628186    1   /data/jaad/image/0316/pedestrian/00...
1   0.171355    0   /data/jaad/image/0316/ped1/0001.png
2   0.628186    1   /data/jaad/image/0316/pedestrian/00...

输出

   pred       true            pedestrian     
0   0.628186   1    /data/jaad/image/0316/pedestrian/00...
1   -1         0    /data/jaad/image/0316/ped1/0001.png 
2   -1         0    /data/jaad/image/0316/ped2/0001.png

0 个答案:

没有答案