I'm trying to use polars dataframe where I would like to select the first and last row per group.