Window Function Syntax In Pyspark