如何在pyspark的函数中使用变量来连接列

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

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 创建示例数据
data = [("John", "Doe", 25), ("Jane", "Smith", 30)]
df = spark.createDataFrame(data, ["first_name", "last_name", "age"])

# 定义要连接的列名
column_name = "full_name"

# 使用变量连接列
df_with_full_name = df.withColumn(column_name, concat(col("first_name"), col("last_name")))

# 显示结果
df_with_full_name.show()