如何在pyspark中根据条件设置新的标志？

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

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

# 创建示例数据集
data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
df = spark.createDataFrame(data, ["Name", "Age"])

# 使用条件语句和函数设置新的标志列
df = df.withColumn("Flag", when(df.Age > 30, "High").otherwise("Low"))

# 显示结果
df.show()