Pandas中groupby后的分组列状态及保留为正式列的方法

当执行df1.groupby(["A"]).sum()时，分组键"A"会被自动设置为df2的索引（Index），而非普通的DataFrame列。所以在显示df2时，索引会被展示出来，但df2.columns仅返回DataFrame的普通列索引，因此找不到"A"。

有两种常用方式可以将"A"保留为正式列：

• 在分组时指定as_index=False参数，直接让分组键保留为列：

  df2 = df1.groupby(["A"], as_index=False).sum()
  

• 如果已经生成了df2，可以通过reset_index()方法将索引转换为普通列：

  df2 = df2.reset_index()
  

内容的提问来源于stack exchange，提问作者Astral