Skip to main content
 首页 » 编程设计

python之Pandas :GroupBy 到 DataFrame

2025年05月04日10zlslch

有一个非常流行的 S.O.有关 groupby 到数据框的问题,请参阅 here .不幸的是,我认为这个特定用例不是最有用的。

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

例如

     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) 

但事实并非如此。

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

请您参考如下方法:

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