Создавать столбцы из pandas groupby

Question

Я пытаюсь посчитать разницу в рейтинге по полу для каждого мова ie. (Набор данных IMDB)
Это метод группировки, который я использовал:

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

И голова:

Я хотел бы создать новый фрейм данных с 2 столбцами рейтинга для каждого mov ie - для мужчин и женщин.
Например, первая строка будет иметь вид:

Спасибо!

Answer 1

Вы можете unstack:

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

И вы должны получить F,M в качестве имен столбцов. Переименуйте как вы wi sh.