python - groupby 的代码优化

Question

我有下面的代码，基本上执行分组操作，然后进行求和。

grouped = df.groupby(by=['Cabin'], as_index=False)['Fare'].sum()

然后我重命名这些列

grouped.columns = ['Cabin', 'testCol']

然后我将“分组”数据框与原始数据框合并以计算聚合。

df2 = df.merge(grouped, on='Cabin')

它的作用是用我的“分组”数据帧中的“testCol”填充我的初始数据帧。

可以优化此代码以适合一行或类似的内容吗？

Answer 1

看来需要GroupBy.transform对于 sum 的新列:

df['testCol'] = df.groupby('Cabin')['Fare'].transform('sum')