https://quadexcel.com/wp/48-databricks-pyspark-find-top-or-bottom-n-rows-per-group/