Python：熊貓數據框將字符串和值轉換為列類別

Question

我想處理下面的分組數據集（導致以下結果的命令是df.groupby(['ids','category'])['counts'].sum() ）

ids|category|counts
1   A        3
2   B        5
3   A        1
    B        1
    C        1
4   C        3

我想要得到的是下面（每個唯一的 id 都有一行，以便我以后可以與另一個數據表合並）：

ids|A_n|B_n|C_n
1   3    0   0
2   0    5   0
3   1    1   1
4   0    0   3

這可能嗎？ 歡迎任何想法，提前謝謝！

Answer 1

只需unstack和fillna

group_df.unstack('category').fillna(0)

輸出

         counts          
category      A    B    C
ids                      
1           3.0  0.0  0.0
2           0.0  5.0  0.0
3           1.0  1.0  1.0
4           0.0  0.0  3.0