python - 如何撤消使用 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())

我得到错误:

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

既没有:

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