python - 如何将一个数据框映射到另一个数据框(python pandas)？

Question

给定这两个数据帧，我如何获得预期的输出数据帧？较长的方法是使用 iloc 循环遍历数据帧的行，然后在将 df2 转换为 后使用 map 函数dict 将 x 和 y 映射到它们的分数。

这看起来很乏味，并且需要很长时间才能在大型数据帧上运行。我希望有一个更清洁的解决方案。

df1:

ID    A    B    C
1     x    x    y
2     y    x    y
3     x    y    y

df2:

ID    score_x    score_y
1          20         30
2          15         17
3          18         22

输出:

ID    A     B     C
1     20    20    30
2     17    15    17
3     18    22    22

注意:数据框会有很多列，并且不仅仅是 x 和 y 作为类别(可能在 20 个类别的区域中)。

谢谢!

Answer 1

使用DataFrame.apply沿着列 Series.map :

df1.set_index('ID', inplace=True)
df2.set_index('ID', inplace=True)
df2.columns = df2.columns.str.split('_').str[-1]

df1 = df1.apply(lambda x: x.map(df2.loc[x.name]), axis=1).reset_index()

print(df1)
   ID   A   B   C
0   1  20  20  30
1   2  17  15  17
2   3  18  22  22

---

print(df2)
     x   y
ID        
1   20  30
2   15  17
3   18  22