Pandas - 帶總和的唯一列數

Question

我有以下數據框：

   x_1  x_2      x_3    x_4        x_5  ID  cost 
0  159  xyz  883nne2  28h93     lightz  10     2       
1  159  xyz  883nne2  28h93     lightz  10     5       
2  354  abc  94mmm4k             heavy  15     2       
3  354  abc  94mmm4k             heavy  15     1       
4  354  abc  94mmm4k   455h             15     7       
5  354  abc  94mmm4k             super   3     2       
6  354  abc  94mmm4k    53g             10     4   
7  354  abc  94mmm4k    53g             10     4   

我想找到列 x_1、x_2、x_3、x_4、x_5、ID 的每個唯一組合的計數，並獲得這些計數的總和，然后應用在成本列中找到的值來獲得每個唯一組合的總和。

希望最終結果如下所示：

   x_1  x_2      x_3    x_4        x_5  ID   count sum
0  159  xyz  883nne2  28h93     lightz  10       2   7      
1  354  abc  94mmm4k             heavy  15       2   3      
2  354  abc  94mmm4k   455h             15       1   7
3  354  abc  94mmm4k             super   3       1   2
4  354  abc  94mmm4k    53g             10       2   8

Answer 1

讓我們用agg做groupby

s=df.groupby(['x_1','x_2','x_3','x_4','x_5','ID'],as_index=False)['cost'].agg(['count','sum'])