把名字在GROUPBY大熊貓聚集funtion

Question

我想一個名字添加到總和列，並把它稱爲「my_sum」 [像「爲」 SQL總和（年齡）爲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