Panda dataframe groupby и присоединяется к столбцам

Question

Есть ли простой способ сделать следующее: у меня есть фрейм данных panda со следующими столбцами: дата (по дням), компания, продажи. Я хочу создать dataframe с продажами для каждой компании в качестве столбца; то есть новые столбцы должны быть: дата, продажа компании 1, продажа компании 2, продажа компании 3. Я хотел бы назвать столбцы по имени в столбце компании.

Я сделал: df.groupby(['company']) и думал, что могу использовать внешнее соединение для каждого элемента в группе. Но это кажется неуместным? Кажется, должен быть более простой способ сделать это? Спасибо! Любая помощь приветствуется.

Примером может служить:

date | company | sales 

1/1/2017 | AAA | 100 

1/1/2017 | BBB | 200 

1/1/2017 | CCC | 300 

И я хочу, чтобы преобразовать его в:

date | AAA | BBB | CCC 

1/1/2017 | 100 | 200 | 300 

Answer 1

вариант 1
set_index + unstack

df.set_index(['date', 'company']).sales.unstack().reset_index() 

company  date AAA BBB CCC 
0  1/1/2017 100 200 300 

вариант 2
pivot

df.pivot('date', 'company', 'sales').reset_index() 

company  date AAA BBB CCC 
0  1/1/2017 100 200 300