Лучший способ добавить групповые итоги в фрейм данных в Pandas

У меня есть простая задача, и мне интересно, есть ли лучший/более эффективный способ сделать это. У меня есть кадр данных, который выглядит так:

  Group  Score  Count
0     A      5    100
1     A      1     50
2     A      3      5
3     B      1     40
4     B      2     20
5     B      1     60

И я хочу добавить столбец, содержащий значение общего количества групп:

  Group  Score  Count  TotalCount
0     A      5    100         155
1     A      1     50         155
2     A      3      5         155
3     B      1     40         120
4     B      2     20         120
5     B      1     60         120

Как я это сделал:

Grouped=df.groupby('Group')['Count'].sum().reset_index()
Grouped=Grouped.rename(columns={'Count':'TotalCount'})

df=pd.merge(df, Grouped, on='Group', how='left')

Есть ли лучший/более чистый способ добавить эти значения непосредственно в фрейм данных?

Спасибо за помощь.


person AJG519    schedule 20.06.2015    source источник


Ответы (1)