熊猫汇总两列的总和并将其作为一列

Question

我正在寻找分组的解决方案，然后在 pandas 数据框中找到两列的总和并显示为一列。

每行的净值和总和列，并添加一个新列“总计”作为两者的总和。

示例数据集如下

    Name  state    Net  Gross
    A1    TN       100     60
    A2    AP       200     40
    A3    KAR      300     -10 
    A1    TN       10      60
    A2    AP       20      40
    A3    KAR      10     -10 

预期输出

    Name  state    Total Amount
    A1    TN       230     
    A2    AP       300     
    A3    KAR      290      

我已经尝试了如下两个步骤

df2 = df1.groupby(['Name', 'State', ]).agg({'Net': 'sum', 'Gross':'sum'})
df2['Total Amount'] = df2['Net'] + df2['Gross']

有没有办法一步完成。谢谢。

Answer 1

一步我们可以先做melt

df.melt(['Name','state']).groupby(['Name','state']).value.sum().reset_index()
Out[56]: 
  Name state  value
0   A1    TN    230
1   A2    AP    300
2   A3   KAR    290