在 Pandas 中，如何基于相同的 ID 遍历两个数据帧

Question

熊猫

我有两个数据框并想做一个嵌套循环。

我想迭代 df1 中的每一行并选择 col1 (id) 和 col2。

然后，它将获取 ID 并遍历 df2 并检查该行是否具有相同的 ID，然后将 df1 中的列与 df2 中的列进行比较

如果 df1 中的 col2 匹配 df2 中的 col3，它将返回 True 并将其附加到 df1 的行。

df1
col1    col2   col3   col3
 01      A     S      True
 02      D     F      True
 03      Z     B      False

df2
col1    col2   col3
 01      A      A
 02      B      A
 02      D      F
 02      C      D
 02      D      V
 03      X      W
 03      E      X

Answer 1

设置

a = df1[['col1', 'col2']].values
b = df2[['col1', 'col2']].values

---

通过any 和all 使用广播：

(a == b[:, None]).any(0).all(1)

array([ True,  True, False])

Answer 2

IIUC 使用 tuple 和 isin

df1[['col1','col2']].apply(tuple,1).isin(df2[['col1','col3']].apply(tuple,1))
Out[1051]: 
0     True
1     True
2    False
dtype: bool