如何消除 pandas groupby 中聚合列的附加行

Question

我正在尝试对字段进行一些聚合，并且在聚合列上获得了额外的行-在本例中为“总和”。 我该如何摆脱这个？ 这使得后续处理很难处理这个dataframe

import pandas as pd
Grade=[['A',1],['A',2],['A',10],['B',4],['B',2],['C',3],['D',10],['D',5],['D',1]]
Grade_df=pd.DataFrame(Grade,columns=['grade','count'])    
Grade_Aggregate=Grade_df.groupby(['grade']).agg({'count':['sum']}).reset_index() 

Grade_Aggregate

  grade count
          sum
0     A    13
1     B     6
2     C     3
3     D    16

Answer 1

如果要删除“总和”：

Grade_Aggregate.columns=[i[0] for i in Grade_Aggregate.columns]

如果您希望将其附加到列名：

Grade_Aggregate.columns=["_".join(i) for i in Grade_Aggregate.columns]

Answer 2

您可以在这里直接使用Groupby.sum 。

Grade_df.groupby('grade', as_index=False)['count'].sum()

  grade  count
0     A     13
1     B      6
2     C      3
3     D     16