pandas - 如何合并来自两个不同 DataFrame 的相同列

Question

我有两个数据框 A 和 B。两者都有相同的 4 列。我想合并两个数据框，如果前三列值匹配，则合并 id 值（这是一个 jasonb 数组）

样本数据：

df_A

name     age    zip      id
abc      25     11111    ["2722", "2855", "3583"]

df_B

name     age    zip      id
abc      25     11111    ["123", "234"]

我希望最终输出看起来像

最终输出：

name     age    zip      id
----------------------------------------------------------------
abc      25     11111    ["2722", "2855", "3583", "123", "234"]

标签： pandas

Answer 1

一种快速的解决方案是

l=['name','age','zip']
df=(df1.set_index(l)+df2.set_index(l)).reset_index()