Pandas DataFrame 分组并求和到新 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 这个文件里。这样你就可以看到两个列，一个是办公室，另一个是费用的总和。