python - 从数据框中选择行，其中两个元素在另一个数据框中的单行上匹配

Question

我正在尝试创建一个新的数据框，其中包含数据框 A 中的所有行，其中 x 列和 y 列的值都与数据框 B 中的一行匹配。

我尝试从数据帧 B 中查找行的索引，其中只有 x 列的值匹配，以尝试查看是否可以匹配这些索引上 y 的值，但我无法检索索引。还尝试了 iterrows 来匹配这两个条件，但同样不成功。

A={'x':[1,2,3,4],'y':[5,6,7,8]}
B={'x':[1,2,3,10],'y':[5,6,90,8]}

我想要返回代码

{'x':[1,2],'y':[5,6]}

Answer 1

假设您有 2 个数据框:

A=pd.DataFrame({'x':[1,2,3,4],'y':[5,6,7,8]})
B=pd.DataFrame({'x':[1,2,3,10],'y':[5,6,90,8]})

然后你可以这样做:

idx = A.x == B.x
idy = A.y == B.y
ID = idx&idy
A[ID]
    x   y
0   1   5
1   2   6

或者，您可以将其转换为所需的输出:

A[ID].to_dict("list")
{'x': [1, 2], 'y': [5, 6]}