Pandas.DataFrame.groupby.sum() 删除列

Question

I have the following dataset:

Assignment	Reference	Amount	Order reason
BB017648	90317000	1,579.54	PEN
BB017648	90748514	3,999.00
BB017648	90317000	540.21
BB001947	90464822	33,003.89	PEN
BB017244	90687323	10.16	REJ

I would like to perform a "pivot table like" aggregation on the column "Reference". I tried this using the following code:

import pandas as pd

wb = pd.read_excel("file.XLSX")
wb = wb.groupby("Reference").sum()

However, the result drops the columns "Order Reason" and "Assignment" and only shows the Reference and Amount.

Is there a way to prevent this?

Answer 1

You can groupby more than one column. Something like this should get what you want?

wb.groupby(["Reference", "Order reason", "Assignment"]).sum()