合并具有多个键列的数据框

Question

我想合并这个数据框：

import pandas as pd
import numpy as np

df1 = pd.DataFrame([[1,10,100],[2,20,np.nan],[3,30,300]], columns=["A","B","C"])
df1
   A   B    C
0  1  10  100
1  2  20  NaN
2  3  30  300

这个：

df2 = pd.DataFrame([[1,422],[10,72],[2,278],[300,198]], columns=["ID","Value"])
df2
    ID  Value
0    1    422
1   10     72
2    2    278
3  300    198

获得输出：

df_output = pd.DataFrame([[1,10,100,422],[1,10,100,72],[2,20,200,278],[3,30,300,198]], columns=["A","B","C","Value"])
df_output 
   A   B    C  Value
0  1  10  100    422
1  1  10  100     72
2  2  20  NaN    278
3  3  30  300    198

想法是，对于 df2，键列是“ID”，而对于 df1，我们有 3 个可能的键列 [“A”，“B”，“C”]。

请注意，为了简单起见，df2 中的数字选择成这样，实际中可以包括随机数。

如何执行这样的合并？谢谢！

Answer 1

IIUC，你需要双merge/join.

首先，melt df1 获取单个列，同时保留索引。然后 merge 得到比赛。最后join到原始DataFrame。

s = (df1
     .reset_index().melt(id_vars='index')
     .merge(df2, left_on='value', right_on='ID')
     .set_index('index')['Value']
    )

# index
# 0    422
# 1    278
# 0     72
# 2    198
# Name: Value, dtype: int64

df_output = df1.join(s)

输出：

   A   B      C  Value
0  1  10  100.0    422
0  1  10  100.0     72
1  2  20    NaN    278
2  3  30  300.0    198

替代 stack + map:

s = df1.stack().droplevel(1).map(df2.set_index('ID')['Value']).dropna()
df_output = df1.join(s.rename('Value'))

合并具有多个键列的数据框

Merging dataframes with multiple key columns

python

pandas

merge