Grupo de chispa por alias

from pyspark.sql.functions import max

df.groupBy('column').agg(max("diff").alias("maxDiff"))
Real Rocket Raccoon