Pyspark Groupby Get First, This is my code.