Pandas DataFrame Groupby和Sum Into New DataFrame

Question

我在数据框中有两列：

office  cost
ABC 20
XYZ 10
ABC 100
XYZ 50

grouped = df.groupby(['office'])
test = grouped.aggregate(np.sum)

当我将分组或测试加载到csv文件中时，我只看到了成本之和 有没有办法可以将列办公室和成本总和加载到新的数据框中，然后加载到csv中？

ABC 120
XYZ  60

Answer 1

test.to_csv('/tmp/out')写道

office,cost
ABC,120
XYZ,60

到/tmp/out。这显示了列，办公室和成本之和。