python-2.7 - 如何选择要在其上取消堆叠数据框的列

Question

我有一个数据集，看起来像:

UniqueID CategoryType Value
   A         Cat1      apple
   A         Cat2      banana
   B         Cat1      orange
   C         Cat2      news
   D         Cat1      orange
   D         Cat2      blue

我希望它看起来像:

UniqueID Cat1 Cat2
   A     apple banana
   B     orange
   C           news
   D     orange blue

我试过使用 unstack，但无法获得正确的索引集或其他东西。

谢谢

Answer 1

大部分工作是用

df.set_index(['UniqueID', 'CategoryType']).Value.unstack(fill_value='')

CategoryType    Cat1    Cat2
UniqueID                    
A              apple  banana
B             orange        
C                       news
D             orange    blue

我们可以得到其余的格式

df.set_index(['UniqueID', 'CategoryType']).Value.unstack(fill_value='') \
    .rename_axis(None, 1).reset_index()

  UniqueID    Cat1    Cat2
0        A   apple  banana
1        B  orange        
2        C            news
3        D  orange    blue