组的总和,但对熊猫中的每一行保持相同的值

问题描述 投票:0回答:1

如何使用熊猫解决此链接Sum of group but keep the same value for each row in r中的相同问题?

我可以为每个组分别生成单独的df,然后将生成的df与原始合并。

pandas
1个回答
0
投票

您可以如下使用groupbytransform来获取输出。

df['sumx']=df.groupby(['ID', 'Group'],sort=False)['x'].transform(sum)
df['sumy']=df.groupby(['ID', 'Group'],sort=False)['y'].transform(sum)
df

输出

ID  Group   x   y   sumx    sumy
1   1   1   1   12  3   25
2   1   1   2   13  3   25
3   1   2   3   14  3   14
4   3   1   4   15  15  48
5   3   1   5   16  15  48
6   3   1   6   17  15  48
7   3   2   7   18  15  37
8   3   2   8   19  15  37
9   4   1   9   20  30  63
10  4   1   10  21  30  63
11  4   1   11  22  30  63
12  4   2   12  23  12  23
© www.soinside.com 2019 - 2024. All rights reserved.