熊貓DataFrame Groupby和總和成新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 。 這將同時顯示列，辦公室和費用總和。