Skip to main content
 首页 » 编程设计

python之如何撤消使用 groupby 转换的 DataFrame 并将其分配给新变量

2025年05月04日14虾米姐

分配给新对象时如何撤消groupby

df = pd.DataFrame({'A': [1, 2, 1, 2, 1, 2, 3], 'B': [1, 1, 1, 2, 2, 2, 2]}) 
AA = df.groupby('A') 
display(AA.unstack()) 

我得到错误:

Cannot access callable attribute 'reset_index' of 'DataFrameGroupBy' objects, try using the 'apply' method

既没有:

display(AA.reset_index()) 

请您参考如下方法:

一个简单的 head() 将返回整个内容

AA = df.groupby('A') 
df_back = AA.head(AA.ngroup().size) 
 
Out[326]: 
   A  B 
0  1  1 
1  2  1 
2  1  1 
3  2  2 
4  1  2 
5  2  2 
6  3  2