How can I resolve TypeError: Column is not iterable in my PySpark DataFrame operation