如何从groupby中获取数据框

Question

我正在做groupby练习。但它返回字典而不是数据帧。即使运气不好，我也从 Stack Overflow 上找到了一些解决方案。

我的代码：

result[comNewColName] = sourceDF.groupby(context, as_index=False)[aggColumn].agg(aggOperation).reset_index()

我尝试过：

result[comNewColName] = sourceDF.groupby(context)[aggColumn].agg(aggOperation).reset_index()

在

result[comNewColName] = sourceDF.groupby(context, as_index=False)[aggColumn].agg(aggOperation)

所有三种情况下，我只得到字典。但我应该在这里获取数据框

：

comNewColName = "totalAmount"
context =['clientCode']
aggColumn = 'amount'
aggOperation = 'sum'

Answer 1

如果需要由聚合值创建的新列，请使用 GroupBy.transform，但分配给 sourceDF：

sourceDF[comNewColName] = sourceDF.groupby(context)[aggColumn].transform(aggOperation)

您的解决方案返回 DataFrame：

df = sourceDF.groupby(context)[aggColumn].agg(aggOperation).reset_index()
print (type(df))