python - 熊猫：groupby后如何获得特定的组？

Question

我想按一个名为“A”的列对数据框进行分组，并检查一个特定的组。

grouped = df.groupby('A', sort=False)

但是，我不知道如何访问一个组，例如，我希望

grouped.first() 

会给我第一组

或者

grouped['foo'] 

会给我所在的组A=='foo'。

然而，Pandas 并不是这样工作的。

我在网上找不到类似的例子。

Answer 1

试试：grouped.get_group('foo')，这就是你所需要的。

Answer 2

from io import StringIO # from StringIO... if python 2.X
import pandas
data = pandas.read_csv(StringIO("""\
area,core,stratum,conc,qual
A,1,a,8.40,=
A,1,b,3.65,=
A,2,a,10.00,=
A,2,b,4.00,ND
A,3,a,6.64,=
A,3,b,4.96,=
"""), index_col=[0,1,2])

groups = data.groupby(level=['area', 'stratum'])
groups.get_group(('A', 'a')) # make sure it's a tuple

                    conc qual
area core stratum            
A    1    a         8.40    =
     2    a        10.00    =
     3    a         6.64    =