Selecting a subset of a Pandas DataFrame

Question

I have the following DateFrame:

 record_id  month  day  year  plot species  sex  wgt 
    33320      33321      1   12  2002     1      DM    M   44
    33321      33322      1   12  2002     1      DO    M   58
    ...          ...      ...  ...  ...    ...    ...   ... ...

I want to display columns from year to wgt with values equal 27. I have done it on two lines:

df_slice =  df[df.year == 27]
df_slice = df_slice.ix[:,'year':]

Is there any way to reduce it to one line?

jezrael · Accepted Answer · 2016-08-01 15:04:13Z

2

You can use ix:

print (df.ix[df.year == 27, 'year':])

Sample (value 2001 was added):

print (df)
   record     id  month  day  year  plot species sex  wgt
0   33320  33321      1   12  2001     1      DM   M   44
1   33321  33322      1   12  2002     1      DO   M   58

print (df.ix[df.year == 2001, 'year':])
   year  plot species sex  wgt
0  2001     1      DM   M   44

answered Aug 1, 2016 at 15:04

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Monica Over a year ago

It was very helpful. What about if I want to display years: 2001 and 2002 and the same time. I've tried df.ix[df.year == 2001 & df.year == 2001, 'year':], but it's not the right solution

jezrael Over a year ago

You are very close, only add brackets df.ix[(df.year == 2001) & (df.year == 2001), 'year':],

DeepSpace · Accepted Answer · 2016-08-01 15:03:08Z

2

Is there any way to reduce it to one line?

You can easily combine the 2 lines into 1:

df_slice = df[df.year == 27].ix[:,'year':]

answered Aug 1, 2016 at 15:03

DeepSpace

82.1k12 gold badges119 silver badges166 bronze badges

Collectives™ on Stack Overflow

Selecting a subset of a Pandas DataFrame

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related