Python Pandas - How to select only the first N rows for each unique value of a column

Question

I have a DataFrame where I need to select no more than 3 rows having a certain value in the column Company:

Which should become:

I tried with the .groupby().size() but I am missing something for sure.

user_9167382 · Answer

Simply, Option 1:

df.groupby('COMPANY').head(3)

Option 2: You could loop through all the unique values in the column and print the output:

for i in df['COMPANY'].unique():
    x = df[df['COMPANY']==i].head(3)
    print(x)

user_3599283 · Answer

I believe to filter based on specific company:

df[df.COMPANY == 'xxxx'].head(3)