extract the first four rows of the column `id` from a pandas dataframe `df`,df.groupby('ID').head(4)