python - pandas pivot - 总结两个字段，依赖于和独立于定义的列

Question

假设有一个像这样的数据框:

Name    |    Type    |    Year    |    Qty   | Amount
Mike        buy          2019        5          100          
James       return       2019        10         150
Mike        return       2019        15         200
James       buy          2019        20         250

我有兴趣按类型对Amount求和，然后对Qty求和而不考虑type，例如:

Name  |  Year  |  buy  |  return  |  Qty
Mike     2019    100         200     20
James    2019    250         150     30   

我怎样才能以最快的方式实现这一点？

我尝试过:df=pd.pivot_table(df, values=['Amount,'Qty'], index=['Name','Type','Year'], columns=['Type'], aggfunc='sum ').reset_index()

但是，这会通过buy 和return 汇总Qty，而不是将Qty 相加而不考虑输入。

Answer 1

分组不同，所以concat两个groupby:

pd.concat([df.groupby(['Name', 'Year', 'Type']).Amount.sum().unstack(-1),
           df.groupby(['Name', 'Year',]).Qty.sum()], 
          axis=1)

            buy  return  Qty
Name  Year                  
James 2019  250     150   30
Mike  2019  100     200   20