python - 过滤 Pandas 数据框聚合

Question

我有一个分组的 pandas 数据框，然后执行聚合计算以获得平均值:

grouped = df.groupby(['year_month', 'company'])
means = grouped.agg({'size':['mean']})

这给了我一个数据框，但我似乎无法将其过滤到我想要的特定公司和 year_month:

means[(means['year_month']=='201412')]

给我一​​个 KeyError

Answer 1

问题是您根据 'year_month' 和 'company' 进行分组。因此，在 means DataFrame 中，year_month 和 company 将成为索引 (MutliIndex) 的一部分。您不能像访问其他列那样访问它们。

执行此操作的一种方法是获取索引的 'year_month' 级别的值。示例 -

means.loc[means.index.get_level_values('year_month') == '201412']

演示 -

In [38]: df
Out[38]:
   A  B   C
0  1  2  10
1  3  4  11
2  5  6  12
3  1  7  13
4  2  8  14
5  1  9  15

In [39]: means = df.groupby(['A','B']).mean()

In [40]: means
Out[40]:
      C
A B
1 2  10
  7  13
  9  15
2 8  14
3 4  11
5 6  12

In [41]: means.loc[means.index.get_level_values('A') == 1]
Out[41]:
      C
A B
1 2  10
  7  13
  9  15