python-3.x - 如何在分组和聚合操作后访问 Pandas DataFrame 中的某一行

Question

我想在 group by 和聚合操作之后访问 Pandas DataFrame 中的一行。

import pandas as pd
df = pd.DataFrame({'A' : ['foo', 'bar', 'foo', 'bar',
                           'foo', 'bar'],
                    'B' : [1, 2, 3, 4, 5, 6],
                    'C' : [2.0, 5., 8., 1., 2., 9.]})

grouped = df.groupby('A').agg({"C":{"size","mean"}})

现在我想访问 A 的值为“foo”的行。
当我尝试使用 grouped[grouped["A"]=="foo"] 时, 我收到一条错误消息 KeyError: 'A'
具体来说，我想要“foo”的大小。

在网上搜索的时候，看到了几篇关于multiIndex的帖子。但我无法让它工作。

这似乎是一个微不足道的问题。我是 Pandas 的新手，发现它有点难以理解。

Answer 1

您应该使用 loc :

grouped.loc['foo']
Out[1]: 
C  size    3.0
   mean    4.0
Name: foo, dtype: float64