如何重命名spark dataframe中的列？

# 导入必要的库
from pyspark.sql import SparkSession

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

# 创建示例DataFrame
data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
df = spark.createDataFrame(data, ["name", "age"])

# 重命名列
df_renamed = df.withColumnRenamed("age", "new_age")

# 显示重命名后的DataFrame
df_renamed.show()

+-------+-------+
|   name|new_age|
+-------+-------+
|  Alice|     25|
|    Bob|     30|
|Charlie|     35|
+-------+-------+