Как сгруппировать по значению полей в Python Pandas?

Question

[Алексей Меликов]

Привет. С Pandas только начинаю работать, поэтому не совсем понимаю как сделать запрос (один или больше?), чтобы получить следующие данные.

Есть такой Dataframe:

id date event city
1 ... e1 city1
2 ... e1 city1
3 ... e2 city1
4 ... e2 city2
5 ... e1 city3

Надо получить таблицу:

city e1 e2
city1 2 1
city2 0 1
city3 1 0

Любая помощь приветствуется. Спасибо.

Answer 1

[Andrey Dugin]

pd.get_dummies(df[['event', 'city']], prefix='', prefix_sep='', columns=['event']).groupby('city').sum()

Можно ещё в конце добавить .reset_index()

Comments

[Алексей Меликов]

Великолепно! На самом деле и ожидал чего-то подобного, знаю каким богатым функционалом обладает Pandas. Мое решение было таким (по аналогии с JOIN в SQL):

pandas.merge(
    data_events[['event', 'city']] \
        .query('event == "e1"') \
        .groupby(['city']) \
        .count(),
    data_events[['event','city']] \
        .query('event == "e2"') \
        .groupby(['city']) \
        .count(),
    on='city',
    how='left'
)

Но Ваше решение гораздо лучше. Спасибо!