如何使用Pyspark中的when语句和array_contains根据条件创建新列？

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

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

# 创建示例数据
data = [("Alice", ["apple", "banana", "orange"]),
        ("Bob", ["grape", "watermelon"]),
        ("Charlie", ["apple", "orange"]),
        ("David", ["banana", "grape"]),
        ("Eva", ["apple", "banana"])]

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

# 使用when语句和array_contains函数创建新列
df = df.withColumn("has_apple", when(array_contains(df.fruits, "apple"), "Yes").otherwise("No"))

# 显示结果
df.show()

+-------+------------------+---------+
|   name|            fruits|has_apple|
+-------+------------------+---------+
|  Alice|[apple, banana, o]|      Yes|
|    Bob|[grape, watermelon]|       No|
|Charlie|  [apple, orange, ]|      Yes|
|  David|   [banana, grape]|       No|
|    Eva|  [apple, banana,]|      Yes|
+-------+------------------+---------+