熊貓數據框分組並加入列

Question

是否有一種簡單的方法可以執行以下操作：我有一個包含以下列的熊貓數據框：日期（按天）、公司、銷售。 我想創建一個數據框，將每個公司的銷售額作為一列； 即，新列應為：日期、公司 1 銷售、公司 2 銷售、公司 3 銷售。 我想按公司列中的名稱命名列。

我做了： df.groupby(['company'])並認為我可以為組中的每個元素進行外部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