将列(变量)列表连接到一个新的列dataframe pyspark中

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

spark = SparkSession.builder.getOrCreate()

data = [("John", 25), ("Jane", 30), ("Tom", 35)]
df = spark.createDataFrame(data, ["Name", "Age"])
df.show()

+----+---+
|Name|Age|
+----+---+
|John| 25|
|Jane| 30|
| Tom| 35|
+----+---+

columns = ["Salary", "Department"]

new_df = df.withColumn("NewColumn", col(columns[0]))
for column in columns[1:]:
    new_df = new_df.withColumn(column, col(column))
new_df.show()

+----+---+-------+----------+
|Name|Age| Salary|Department|
+----+---+-------+----------+
|John| 25|   null|      null|
|Jane| 30|   null|      null|
| Tom| 35|   null|      null|
+----+---+-------+----------+