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

Question

分配给新对象时如何撤消 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())

我收到错误：

无法访问“DataFrameGroupBy”对象的可调用属性“reset_index”，请尝试使用“apply”方法

既不具有：

display(AA.reset_index())

Answer 1

一个简单的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