Есть ли способ сделать колонку для каждой категории в каждой группе?

Question

Есть два столбца, оба являются категориальными. Я хочу, чтобы группировка по первым столбцам позволила age_group, а мой другой столбец Engagement_category должен создать новый столбец по каждому элементу.

Я сделал группирование по первому столбцу age_group.

| age_group    | Engagement_category |
|--------------|---------------------|
| 21-26        | Not Engaged         |
| 27-32        | Disengaged          |
| 33-38        | Engaged             |
| 39-44        | Disengaged          |
| 45-50        | Not Engaged         |
| 50 and Above | Engaged             |

group = df.groupby('age_group')

Требуется следующий вывод:

| age_group | Engaged | Nearly Engaged | Not Engaged | Disengaged |
|-----------|---------|----------------|-------------|------------|
| 21-26     | 3       | 4              | 1           | 1          |
| 27-32     | 4       | 0              | 4           | 0          |
| 33-38     | 2       | 0              | 1           | 1          |

Спасибо.

Answer 1

Вы хотите сгруппировать по первым двум столбцам, а затем использовать функцию «COUNT» для агрегирования по всему фрейму данных. Попробуйте это:

df.groupby(['Age', 'Engagement_category']).agg(['count'])