在两个数据框之间找到相等的列

Question

我有两个pandas数据框，a和b：

a1   a2   a3   a4   a5   a6   a7
1    3    4    5    3    4    5
0    2    0    3    0    2    1
2    5    6    5    2    1    2

和

b1   b2   b3   b4   b5   b6   b7
3    5    4    5    1    4    3
0    1    2    3    0    0    2
2    2    1    5    2    6    5

这两个数据帧包含完全相同的数据，但顺序不同且列名不同。基于这两个数据帧的数字，我希望能够将每个列名在匹配a到每个列名b。

很难简单地将的第一行a与的第一行进行比较，b因为存在重复的值（例如两者都具有a4且a7具有值），5因此不可能立即将它们与b2或匹配b4。

做这个的最好方式是什么？

Answer 1

这是一种利用广播检查两个数据帧之间是否相等并all利用结果检查所有行匹配的一种方法。然后，我们可以从np.where
（带有@piR的贡献）的结果中获得两个数据框的列名称的索引数组：

i, j = np.where((a.values[:,None] == b.values[:,:,None]).all(axis=0))
dict(zip(a.columns[j], b.columns[i]))
# {'a7': 'b2', 'a6': 'b3', 'a4': 'b4', 'a2': 'b7'}