Pyspark Select Row With Max Value