python - Pandas:如何在groupby之后将值分配给组的第一行？

Question

我试图将值分配给 组的第一行在groupby之后？
输入:

ID Num
Aa 100
Bb 200
Cc 300
Bb 400
Aa 500

输出:

ID Num Sum
Aa 100 600
Aa 500
Bb 200 600
Bb 400
Cc300 300

代码:

base_df['Sum'] = base_df.groupby(['ID'])['Num'].transform('sum')

但是，此代码会将值 'Sum' 分配给 每行在组中:
我的输出:

ID Num Sum
Aa 100 600
Aa 500 600
Bb 200 600
Bb 400600
Cc 300 300

Answer 1

您可以使用 duplicated 进行分配:

s = df.groupby(['ID'])['Num'].transform('sum')
df.loc[~df.duplicated("ID"), "Sum"] = s
print (df)

   ID  Num    Sum
0  Aa  100  600.0
1  Bb  200  600.0
2  Cc  300  300.0
3  Bb  400    NaN
4  Aa  500    NaN