使用来自数据帧的列向量值和来自另一个数据帧的列表列合并 2 个数据帧

Question

假设我有两个数据框，

df1.head()
AH1
A
B
C
D
F

第二个数据框df2.head()

AH1 BC
A,B SAP
C,D XY
A,B,C   BMZ
A,F Zoom

我需要基于相同的值合并两个数据框df1 和df2 中的列AH1。最后，我的目标是拥有这样的东西，

output
    AH1 BC
    A   SAP, BMZ,Zoom
    B   SAP,BMZ
    C   XY,BMZ
    D   XY
    F   Zoom

感谢任何帮助或建议。我知道pandas.merge 不会出现在列向量和列表列之间。

Answer 1

你可以做一个explode:

(df2.assign(AH1=df2.AH1.str.split(','))
    .explode('AH1')
    .groupby('AH1')
    ['BC'].agg(list)
)

输出：

AH1
A    [SAP, BMZ, Zoom]
B          [SAP, BMZ]
C           [XY, BMZ]
D                [XY]
F              [Zoom]
Name: BC, dtype: object

或者用agg(', '.join)替换agg(list)：

(df2.assign(AH1=df2.AH1.str.split(','))
    .explode('AH1')
    .groupby('AH1')
    ['BC'].agg(', '.join)
)

输出：

AH1
A    SAP, BMZ, Zoom
B          SAP, BMZ
C           XY, BMZ
D                XY
F              Zoom
Name: BC, dtype: object