在pyspark中交换列值

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

spark = SparkSession.builder.getOrCreate()

data = [("John", "Doe", 25), ("Jane", "Smith", 30), ("Tom", "Brown", 35)]
df = spark.createDataFrame(data, ["first_name", "last_name", "age"])

df = df.withColumn("temp", col("first_name"))
df = df.withColumn("first_name", col("last_name"))
df = df.withColumn("last_name", col("temp")).drop("temp")

df.show()