Pandas get topmost n records within each group

by

Did you try

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

Output generated:

       id  value
id             
1  0   1      1
   1   1      2 
2  3   2      1
   4   2      2
3  7   3      1
4  8   4      1

(Keep in mind that you might need to order/sort before, depending on your data)

EDIT: As mentioned by the questioner, use

df.groupby('id').head(2).reset_index(drop=True)

to remove the MultiIndex and flatten the results:

    id  value
0   1      1
1   1      2
2   2      1
3   2      2
4   3      1
5   4      1

More Related Contents:

  1. Select row by max value in group in a pandas dataframe
  2. Find names of top-n highest-value columns in each pandas dataframe row
  3. Select the max row per group – pandas performance issue
  4. how many countries have organised more than one edition of olympic games? using python pandas [closed]
  5. How to sort a dataFrame in python pandas by two or more columns?
  6. pandas get column average/mean
  7. python pandas extract year from datetime: df[‘year’] = df[‘date’].year is not working
  8. How can I strip the whitespace from Pandas DataFrame headers?
  9. Select Pandas rows based on list index
  10. selecting across multiple columns with python pandas?
  11. Problem in combining bar plot and line plot (python)
  12. Bar Chart: How to choose color if value is positive vs value is negative
  13. Python pandas: remove everything after a delimiter in a string
  14. import pandas_datareader gives ImportError: cannot import name ‘is_list_like’
  15. Finding the intersection between two series in Pandas
  16. vectorize conditional assignment in pandas dataframe
  17. Read csv from Google Cloud storage to pandas dataframe
  18. How to set the pandas dataframe data left/right alignment?
  19. Convert numpy type to python
  20. How to get/set a pandas index column title or name?
  21. Pandas: drop columns with all NaN’s
  22. Getting vertical gridlines to appear in line plot in matplotlib
  23. How can I create the minimum size executable with pyinstaller?
  24. dataframe to long format
  25. Can you format pandas integers for display, like `pd.options.display.float_format` for floats?
  26. How to convert string representation of dictionary in Pandas DataFrame to a new columns?
  27. Get the row corresponding to the max in pandas GroupBy [duplicate]
  28. countplot() with frequencies
  29. Horizontal stacked bar chart in Matplotlib
  30. pandas combine two strings ignore nan values

Leave a Comment