https://quadexcel.com/wp/pyspark-interview-questions-select-top-n-rows-from-each-group/