[Pandas] summation based on a column grouping

Import Data

import pandas as pd
 
df = pd.DataFrame([['A',12,0,0],
                   ['A',0,37,0],
                   ['A',0,0,18],
                   ['B',123,0,0],
                   ['B',110,49,61]
                  ], 
                   columns = ['team','value1','value2','value3'])

df

Problem:   Group by field 'team' to sum fields 'value1', 'value2' and 'value3' separately

# df = df.groupby(by=['team'])['value1','value2','value3'].sum().reset_index()

calculated_list = ['value1','value2','value3']
df = df.groupby(by=['team'])[calculated_list].sum().reset_index()

df


groupby grouping aggregation operation https://blog.csdn.net/Hudas/article/details/127961579

Guess you like

Origin blog.csdn.net/Hudas/article/details/130428227