Сводная таблица с несколькими столбцами из Groupby Python

Question

После групповой операции (идентификатор, пол и работа) у меня есть это:

id        geneder      job           salary
22         male       engineer         100
                      doctor           120
                       ...
           female     engineer          90
                       doctor          100
                       ...

23         male       engineer         200
                      doctor           150
                       ...
           female     engineer          80
                       doctor          100

Из этого я хочу создать что-то вроде этого:

id       m_engineer    m_doctor ... f_engineer    f_doctor
22          100          120           90           100
23          200          150           80           100

Я пыталсяс сводными таблицами, но я продолжаю иметь ошибки.Есть идеи?

Answer 1

Вам нужно создать новый столбец, который вы можете поворачивать, что-то вроде

(grouped_df
 .reset_index()
 .assign(gender_job = lambda df: df['gender'].str[0] + '_' + df['job'])
 .pivot('id', 'gender_job', 'salary'))