How to subset a dataset in pandas dataframe?,df.groupby('ID').head(4)