python - Pandas DataFrame 合并求和列

Question

我正在尝试合并两个 DataFrames 求和列值。

>>> print(df1)
   id name  weight
0   1    A       0
1   2    B      10
2   3    C      10

>>> print(df2)
   id name  weight
0   2    B      15
1   3    C      10

在合并公共(public)列中的相似值时，我需要对 weight 值求和。

merge = pd.merge(df1, df2, how='inner')

因此输出将如下所示。

   id name  weight
1   2    B      25
2   3    C      20

Answer 1

如果您想对多个列求和，此解决方案也适用。假设数据框

>>> df1
   id name  weight  height
0   1    A       0       5
1   2    B      10      10
2   3    C      10      15
>>> df2
   id name  weight  height
0   2    B      25      20
1   3    C      20      30

您可以连接它们并按索引列分组。

>>> pd.concat([df1, df2]).groupby(['id', 'name']).sum().reset_index()
   id name  weight  height
0   1    A       0       5
1   2    B      35      30
2   3    C      30      45