将名称放在groupby pandas中的聚合函数中

Question

我想在sum列中添加一个名称并将其命名为'my_sum'[与SQL sum（age）中的“as”一样称为my_sum]

我设法做到了这一点

my_group=df.groupby(['year']).agg({ 'age':{'my_sum': 'sum'}})

你觉得从这开始有一个更简单的方法吗？：

my_group=df.groupby(['year'])['age'].sum()

Answer 1

我认为您可以将reset_index与参数name：

一起使用

my_group=df.groupby(['year'])['age'].sum().reset_index(name='my_sum')

样品：

df = pd.DataFrame({'year':[2000,2001,2000],
                   'age':[40,50,60]})
print (df)
   age  year
0   40  2000
1   50  2001
2   60  2000

my_group=df.groupby(['year'])['age'].sum().reset_index(name='my_sum')
print (my_group)
   year  my_sum
0  2000     100
1  2001      50