如何撤消使用 groupby 转换的数据帧并识别到新变量

Question

当分配给新对象时，如何撤消分组依据

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

Answer 2

您可以将每个组放在一个列表中并像下面这样连接：

pd.concat([g for _,g in AA]).sort_index()

---

A  B
0  1  1
1  2  1
2  1  1
3  2  2
4  1  2
5  2  2
6  3  2