如何在pyspark中将多列合并为一列

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

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

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

# 将"first_name"和"last_name"合并为一个新列"full_name"
df = df.withColumn("full_name", concat(df.first_name, df.last_name))

# 显示结果
df.show()

+----------+---------+---+----------+
|first_name|last_name|age|full_name |
+----------+---------+---+----------+
|John      |Doe      |25 |JohnDoe   |
|Jane      |Smith    |30 |JaneSmith |
+----------+---------+---+----------+