Pyspark Loop Over Rows