I want to generate unique (numerical) labels to each group in my data in pyspark. Similar to group_indices in R: https://dplyr.tidyverse.org/reference/group_indices.html
