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