在熊猫中分组的项目总数和在文件中写入

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使用transform生成的列：

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

然后您可以在上面调用to_csv将其写入文件