python - 比较两个 Python Pandas 数据帧的 2 列并获取公共(public)行

Question

我有 2 个 Dataframe，如下所示:

DF1=
    A    B   C    D
0   AA   BA  KK   0
1   AD   BD  LL   0
2   AF   BF  MM   0

DF2=
    K    L
0   AA   BA
1   AD   BF
2   AF   BF

最后我想得到的是:

DF1=
    A    B   C    D
0   AA   BA  KK   1
1   AD   BD  LL   0
2   AF   BF  MM   1

所以，我想比较两个数据帧，我想看看第一个数据帧(A 列和 B 列)的哪些行与第二个数据帧(K 列和 L 列)的相同，并在列 D 上分配 1第一个数据框。

我可以使用 for 循环，但是对于大量条目来说它会很慢。

任何线索或建议将不胜感激。

Answer 1

如果您重命名 df2 的列，然后您可以逐行比较，这会更容易:

In [35]:

df2.columns = ['A', 'B']
df2
Out[35]:
    A   B
0  AA  BA
1  AD  BF
2  AF  BF
In [38]:

df1['D'] = (df1[['A', 'B']] == df2).all(axis=1).astype(int)
df1
Out[38]:
    A   B   C  D
0  AA  BA  KK  1
1  AD  BD  LL  0
2  AF  BF  MM  1