pandas: select the first couple of rows in each group,"df.groupby('id', as_index=False).head(2)"