Filtering dataframe rows based on count

Question

I have a table like

Fruit   Month
----------------
A   Jan
A   Feb
A   Mar 
B   Sep

I want to filter the table in such a way that I get only top n rows say two for each fruit Example

Fruit   Month
----------------
A   Jan
A   Feb 
B   Sep

rahlf23 · Accepted Answer · 2019-04-02 16:47:14Z

2

You can just use groupby() and head():

df.groupby('Fruit').head(2)

Outputs:

  Fruit Month
0     A   Jan
1     A   Feb
3     B   Sep

answered Apr 2, 2019 at 16:47

rahlf23

9,0494 gold badges30 silver badges57 bronze badges

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

4 Comments

anky Over a year ago

ahh i think i got this question wrong. Guessed its based on months

rahlf23 Over a year ago

Yeah, I wasn't entirely sure what the OP was aiming for, but this seems to be the most simplistic way to achieve their desired output.

Constantine1991 Over a year ago

This works like a charm but the issue I have is it does not maintain the initial order of top two but selects random two. Is there a way to solve that?

rahlf23 Over a year ago

It will not select a random two, can you please include a revised example in your original post with what you are observing?

Collectives™ on Stack Overflow

Filtering dataframe rows based on count

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related