python - 如何访问使用 DataFrame.groupby 创建的 DataFrame 列

Question

在创建一个包含列 'a' 且单元格值重复的 DataFrame 之后:

import pandas as pd
df = pd.DataFrame({'a': [1,2,2,3,3,3,3],
                   'b':[1,2,3,4,5,6,7],
                   'c':[8,9,10,11,12,13,14]})

我继续创建它的副本，在其中合并“a”列中的重复单元格，同时对其他单元格的值求和。我正在使用 groupby 方法来实现这一点，如下所示:

groupDf = df.groupby('a', axis=0).sum()

创建 DataFrame 的合并版本后，我想访问它的列 ['a']:

print(groupDf['a])

我收到 KeyError 通知我 ['a'] 列不存在。使用 groupDf.columns 打印列名仅返回“b”列和“c”列:Index(['b', 'c'], dtype='object')

我应该如何访问“a”列？

Answer 1

在groupby之后，分组的列变成了一个索引，你可以通过调用.index或者reset_index来访问它，然后访问它作为一个普通的列，即以下两种方法:

df.groupby('a', axis=0).sum().reset_index()

#   a   b   c
# 0 1   1   8
# 1 2   5   19
# 2 3   22  50

df.groupby('a', axis=0).sum().index
# Int64Index([1, 2, 3], dtype='int64')