Pyspark Get Max