pyspark如何根据值添加选中的列

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

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

# 创建示例DataFrame
data = [("Alice", 25, "F"),
        ("Bob", 30, "M"),
        ("Charlie", 35, "M")]

df = spark.createDataFrame(data, ["name", "age", "gender"])

# 添加选中的列
df_with_selected_column = df.withColumn("selected", when(col("age") > 30, "Yes").otherwise("No"))

# 显示结果
df_with_selected_column.show()