python - 合并数据帧以包含指示行属于哪些数据帧的列

Question

我有两个 Dataframe(a 是可哈希的，b 是不可哈希的)

   Foo              
    a   b           
A   1   2       
B   1   3       
C   0   4       

 Bar
a   b
A   1   2
D   0   6

如果索引或代码(本例中为 A、B、C、D)相同，则 a 和 b 必须相同。我想知道如何合并这两个数据框，显示列所属的位置，如下所示。

    a   b   Foo   Bar
A   1   2    1     1
B   1   3    1     0
C   0   4    1     0
D   0   6    0     1

获取此数据帧的最有效方法是什么？

Answer 1

您应该能够使用df.merge来做到这一点:

df1['Foo'] = 1
df2['Bar'] = 1
out = df1.merge(df2, on=['a', 'b'], how='outer').fillna(0).astype(int)

print(out)
   a  b  Foo  Bar
0  1  2    1    1
1  1  3    1    0
2  0  4    1    0
3  0  6    0    1