human
How to subset a dataset in pandas dataframe?
assistant
Here is some code to extract the first four rows of the column `ID` from a pandas dataframe `df`

df.groupby('ID').head(4)

#Warning: this code generation is experimental. Please inspect the code for any bugs before executing.