将过滤器应用于Spark中数组列的元素的最简洁方法是什么

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

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

# 创建示例数据
data = [("John", ["apple", "banana", "orange"]),
        ("Mike", ["grape", "kiwi", "mango"]),
        ("Lisa", ["apple", "mango", "pear"])]

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

# 使用explode()函数将数组列拆分为多行
df_exploded = df.select("name", explode("fruits").alias("fruit"))

# 使用filter()函数对拆分后的元素进行过滤
filtered_df = df_exploded.filter(col("fruit") == "apple")

# 显示过滤后的结果
filtered_df.show()