如何从Spark dataframe中的其他列值创建新列？

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

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

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

# 使用withColumn方法创建新列
df_with_new_column = df.withColumn("age_plus_height", col("age") + col("height"))

# 显示结果
df_with_new_column.show()

+-------+---+------+----------------+
|   name|age|height|age_plus_height |
+-------+---+------+----------------+
|  Alice| 25|   160|             185|
|    Bob| 30|   175|             205|
|Charlie| 35|   180|             215|
+-------+---+------+----------------+