熊猫数据框与分配有特定值的重复项合并

时间:2019-06-27 12:14:35

标签: python pandas dataframe merge

我有两个数据框。

data1 = {'col4':['Tom', 'nick', 'krish', 'jack'],
        'col5':[20, 21, 19, 18]}
df1 = pd.DataFrame(data1)
     col4  col5                                                                                                                  
0    Tom    20                                                                                                                   
1   nick    21                                                                                                                   
2  krish    19                                                                                                                   
3   jack    18 


data2 = {'col1':['xxx', 'yyy', 'zzz', 'xxx','yyy', 'xxx', 'yyy', 'zzz'],
        'col2':['Tom', 'nick', 'krish', 'jack','Tom', 'nick', 'krish', 'jack'],
        'col3':[110, 121, 119, 118,141,156,176,186]}
df2 = pd.DataFrame(data2)
  col1   col2  col3                                                                                                              
0  xxx    Tom    110                                                                                                              
1  yyy   nick    121                                                                                                              
2  zzz  krish    119                                                                                                              
3  xxx   jack    118                                                                                                              
4  yyy    Tom    141                                                                                                              
5  xxx   nick    156                                                                                                              
6  yyy  krish    176                                                                                                              
7  zzz   jack    186

我需要在基于col2和col4的df2数据框中包含col5。 col2中重复值的col5值应分配为0(第一个值除外)。例如,Tom正在重复,并且在与df1合并后,仅将Tom(索引0)的第一个值分配给20。 Tom的另一个值应为0(索引4)。
我希望结果数据框列为:

  col1   col2  col3  col5                                                                                                            
0  xxx    Tom   110   20                                                                                                           
1  yyy   nick   121   21                                                                                                          
2  zzz  krish   119   19                                                                                                           
3  xxx   jack   118   18                                                                                                           
4  yyy    Tom   141    0                                                                                                            
5  xxx   nick   156    0                                                                                                          
6  yyy  krish   176    0                                                                                                         
7  zzz   jack   186    0

请帮助我实现这一目标。 预先感谢。

1 个答案:

答案 0 :(得分:2)

使用np.where()series.duplicated()合并然后应用条件:

final = df2.merge(df1, how='left', left_on='col2', right_on='col4')
final.col5 = np.where(final.duplicated(['col2', 'col5']), 0, final.col5)
print(final)

  col1   col2  col3   col4  col5
0  xxx    Tom   110    Tom    20
1  yyy   nick   121   nick    21
2  zzz  krish   119  krish    19
3  xxx   jack   118   jack    18
4  yyy    Tom   141    Tom     0
5  xxx   nick   156   nick     0
6  yyy  krish   176  krish     0
7  zzz   jack   186   jack     0