在本例中，如何基于相同的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