Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Pandas: how to get a particular group after groupby? [duplicate]

I want to group a dataframe by a column, called 'A', and inspect a particular group.

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

However, I don't know how to access a group, for example, I expect that

grouped.first() 

would give me the first group

Or

grouped['foo'] 

would give me the group where A=='foo'.

However, Pandas doesn't work like that.

I couldn't find a similar example online.

like image 763
zer0ne Avatar asked Mar 28 '14 01:03

zer0ne


2 Answers

Try: grouped.get_group('foo'), that is what you need.

like image 174
CT Zhu Avatar answered Nov 06 '22 09:11

CT Zhu


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    =
like image 4
Paul H Avatar answered Nov 06 '22 07:11

Paul H