熊猫:GroupBy到DataFrame

Question

关于groupby到dataframe有一个非常受欢迎的SO问题,请看这里.不幸的是,我不认为这个特定的用例是最有用的.

假设您有一个展平形式的分层数据集:

例如

     key    val 
0    'a'    2
1    'a'    1
2    'b'    3
3    'b'    4

我想做的是将该数据帧转换为此结构

    'a'  'b'
0    2    3
1    1    4

我觉得这会很简单

pd.DataFrame(df.groupby('key').groups)

但事实并非如此.

那么我该如何进行这种转变呢？

Answer 1

df.assign(index=df.groupby('key').cumcount()).pivot('index','key','val')
Out[369]: 
key    'a'  'b'
index          
0        2    3
1        1    4