gpt4 book ai didi

python - 将组总计添加到 Pandas 中的数据框的最佳方法

转载 作者:太空狗 更新时间:2023-10-29 22:09:13 24 4
gpt4 key购买 nike

我有一个简单的任务,我想知道是否有更好/更有效的方法来完成。我有一个看起来像这样的数据框:

  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')

是否有更好/更简洁的方法将这些值直接添加到数据框?

感谢您的帮助。

最佳答案

df['TotalCount'] = df.groupby('Group')['Count'].transform('sum')

讨论了其他一些选项 here .

关于python - 将组总计添加到 Pandas 中的数据框的最佳方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/30949020/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com