将分组后的子级转置为列 pandas/python

Question

gp1.groupby（by=['ID'，'CD']）['BALANCE\u AM'].sum（）

ID          CD
4332        5                    0.0
            58                   0.0
            123              22656.0
756423      47              645087.0
            123             227655.0

我想为每种类型的CD创建列，余额总和是多少

期望输出

ID          5     58     123       47
4332        0     0      22656.0   NaN
756423      NaN   NaN    227655.0  645087.0

Answer 1

如有必要，在列中添加^{}和^{}ID：

df = gp1.groupby(by=['ID', 'CD'])['BALANCE_AM'].sum().unstack().reset_index()

Answer 2

另一种方法是使用pivot_table而不是groupby：

gp1.pivot_table(values='BALANCE_AM', index='ID', columns='CD', aggfunc='sum')