將 groupby 值轉置到 Python Pandas 中的列

Question

我有這樣的 dataframe

ID   Product   Amount
22   product1  $10
22   product2  $20
22   product3  $30
33   product2  $4
33   product3  $5
44   product1  $78
44   product4  $90

我如何將其轉換為 output 以下（產品列值變為新列）

ID  product1 product2 product3 product4
22       $10      $20      $30     
33                 $4       $5
44       $78                        $90

Answer 1

在這里你 go，使用pivot

test_df = pd.read_csv(StringIO("""ID   Product   Amount
22   product1  $10
22   product2  $20
22   product3  $30
33   product2  $4
33   product3  $5
44   product1  $78"""),sep='\s+')

test_df.pivot(index="ID",columns='Product',values='Amount')


Product product1    product2    product3
ID          
22  $10 $20 $30
33  NaN $4  $5
44  $78 NaN NaN