Pyspark Group By Sort