Spark Dataframe Loop Through Rows Python Compiler