【发布时间】:2016-12-18 19:00:24
【问题描述】:
给定两个大数据帧,是否有任何简洁高效的代码(避免直接使用任何for loop)让我获得这两个数据帧的补码?
对我来说最直接的方法是计算union-intersection,如下面的简单示例所示,但我不知道如何用pandas 或np 的优雅语言来实现它
df1= pd.DataFrame({'key1': ['K0', 'K0', 'K1', 'K2'],
'key2': ['K0', 'K1', 'K0', 'K1'],
'A': ['A0', 'A1', 'A2', 'A3'],
'B': ['B0', 'B1', 'B2', 'B3']})
df2= pd.DataFrame({'key1': ['K0', 'K1', 'K1', 'K2'],
'key2': ['K0', 'K0', 'K0', 'K0'],
'C': ['C0', 'C1', 'C2', 'C3'],
'D': ['D0', 'D1', 'D2', 'D3']})
intersection= pd.merge(df1, df2, how='inner',on=['key1', 'key2'])
union=pd.merge(df1, df2, how='outer',on=['key1', 'key2'])
complement=union-intersection
感谢任何cmets和答案
【问题讨论】:
-
@Mathias711,谢谢。要应用他们的解决方案,我需要在
union中找到intersection的索引。然后我可以应用他们的方法来排除以前获得的索引