映射和合并来自另一个数据框的值

Question

1.输入具有随机数值的数据框：

ID  num
a    2
a,b  3
b    1
c,e  4

我有另一个数据框：

ID  name
a    a1
b    b5
c    c4
d    d1
e    e6

2.预期结果：我想在 ID 上将 df1 与 df2 映射并存储为另一列：

ID  num  ID_name
a    2    a1
a,b  3    a1,b5
b    1    b5
c,e  4    c4,e6

3.我试过的代码：

df1["ID_name"] = df1["ID"].map(df2)
df1

但是这些值没有被映射并显示大多数值的 NAN

Answer 1

我们可以使用Series.str.split，然后在元素上使用Series.map和groupby。

df["ID_name"] = (
    df["ID"]
    .str.split(",")
    .explode()
    .map(df2.set_index("ID")["name"])
    .groupby(level=0)
    .agg(",".join)
)

    ID  num ID_name
0    a    2      a1
1  a,b    3   a1,b5
2    b    1      b5
3  c,e    4   c4,e6