在spark dataframe中使用forloop添加新列

from pyspark.sql import SparkSession
from pyspark.sql.functions import col

spark = SparkSession.builder.getOrCreate()

data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
df = spark.createDataFrame(data, ["Name", "Age"])
df.show()

+-------+---+
|   Name|Age|
+-------+---+
|  Alice| 25|
|    Bob| 30|
|Charlie| 35|
+-------+---+

for col_name in ["Salary", "Department", "Location"]:
    df = df.withColumn(col_name, col("Name"))

df.show()

+-------+---+------+----------+--------+
|   Name|Age|Salary|Department|Location|
+-------+---+------+----------+--------+
|  Alice| 25| Alice|     Alice|   Alice|
|    Bob| 30|   Bob|       Bob|     Bob|
|Charlie| 35|Charlie|  Charlie|Charlie|
+-------+---+------+----------+--------+