如何根据Pyspark中的单元格值重命名列？

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

spark = SparkSession.builder.appName("ColumnRenaming").getOrCreate()

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

+-----+---+
| Name|Age|
+-----+---+
| John| 25|
|Alice| 30|
|  Bob| 35|
+-----+---+

new_column_name = "NewName"
df = df.withColumnRenamed("Name", when(col("Name") == "John", new_column_name).otherwise(col("Name")))
df.show()

+-------+---+
|NewName|Age|
+-------+---+
|   John| 25|
|  Alice| 30|
|    Bob| 35|
+-------+---+