Pyspark Loop Through Dataframe