在将垂直数据转换为水平数据的过程中聚合数据（如R的dcast函数）

Question

输入：

x   y   z   a     b     wkn   year   S1   S2
A   1   6   121   1066  18    2015   10   5
A   1   6   121   1066  18    2015   20   6
A   1   6   121   1066  19    2015   5    3

预期输出：

x   y   z    a      b   wkn   S1_2015  S2_2015
A   1   6   121   1066  18      30      11
A   1   6   121   1066  19      5       3

组键=[x，y，z，a，b，wkn]

聚合函数：和

有人能帮我得到预期的产量吗

提前谢谢

Answer 1

假设您的输入在pandas.DataFrame中，您只需执行以下操作：

df.groupby(['x','y','z','a','b','wkn','year]).agg('sum')

这将导致：

                                    S1  S2
x   y   z   a   b    wkn    year        
A   1   6   121 1066  18    2015    30  11
                      19    2015    5   3