以Pandas为单位分组的项目总和，以fi为单位书写

Question

使用的代码：

bytreatment = df.groupby(['QuoteID'])  
df2['numeric_sort_sum']= bytreatment['ComQuotePrice'].sum()

输入文件：

id1 100
id1 200
id3 200
id3 300
id4 400

结果如下：

id1 300
id1 300
id3 500
id3 500
id4 400

对我来说结果是

id1 300
id3 500
id4 400

我需要把这个结果写入文件。。我应该使用循环和索引吗

Answer 1

要添加从groupby生成的列，请使用^{}：

df2['numeric_sort_sum']= bytreatment['ComQuotePrice'].transform('sum')

In [22]:
df['numeric_sort_sum'] = df.groupby('QuoteID').transform('sum')
df

Out[22]:
  QuoteID  ComQuotePrice  numeric_sort_sum
0     id1            100               300
1     id1            200               300
2     id3            200               500
3     id3            300               500
4     id4            400               400

然后可以调用上面的^{}将其写入ti文件