如何在 pandas 中创建 pivot 表

Question

I have a dataframe 'annual_stats' that I need to reshape into a pivot table. Attached you can see what it currently looks like and what it is supposed to look like after reshaping. what the dataframe is currently

what it needs to look like

Answer 1

Here are two possible ways to achieve this,

annual_stats.groupby(['state', 'year'])['salary'].sum().unstack()

pd.pivot_table(annual_stats, values='salary', index='state', columns='year', aggfunc='mean')