如果符合某些条件，则对来自不同数据帧的两个值求和

Question

如果满足某些条件，我想总结两列，每列在不同的框架中

数据帧1：

desk    Type   total_position
desk1   ES     786.0
desk1   ES1    100
desk2   ES1    0
desk2   ES2    10
desk3   ES     0
desk4   ES1    0
desk5   ES     -757

数据帧2：

desk    Type total_position
desk1   ES   -758.0
desk2   ES    0
desk3   ES   -29
desk4   ES    0.0
desk5   ES    786.0

如果第一个数据帧中的类型是“ES”，并且它是同一张桌子，我想对这两个位置求和

我该怎么做

预期答案

desk    Type   total_position
desk1   ES     29
desk2   ES1    0
desk3   ES     -29
desk4   ES1    0
desk5   ES     29

Answer 1

使用where

>>> df['total_position'] = (df.total_position+df1.total_position).where(df.desk.eq(df1.desk) & df.Type.eq('ES'), 0)
>>> df
    desk Type  total_position
0  desk1   ES            28.0
1  desk2  ES1             0.0
2  desk3   ES           -29.0
3  desk4  ES1             0.0
4  desk5   ES            29.0