Create columns from pandas groupby

Question

I'm trying to calculate the rating difference by gender for each movie. (IMDB dataset)
This is the groupby method i've used:

df.groupby(['movie title', 'gender'])['rating'].mean()

And the head:

I'd like to create a new dataframe with 2 columns of rating for each movie - for male and female.
For example, the first row will be like:

Thanks!

Quang Hoang · Accepted Answer · 2020-05-26 22:20:14Z

You can unstack:

(df.groupby(['movie title', 'gender']) ['rating'].mean() .unstack() .reset_index() # turn `movie_title` into a normal column. )

And you should get F,M as column names. Rename as you wish.

1 Answer 1