如何从pyspark dataframe中查询/提取数组元素

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

spark = SparkSession.builder.getOrCreate()

data = [("Alice", [1, 2, 3]), ("Bob", [4, 5, 6]), ("Charlie", [7, 8, 9])]
df = spark.createDataFrame(data, ["Name", "Numbers"])
df.show()

+-------+---------+
|   Name|  Numbers|
+-------+---------+
|  Alice|[1, 2, 3]|
|    Bob|[4, 5, 6]|
|Charlie|[7, 8, 9]|
+-------+---------+

df_exploded = df.select(col("Name"), explode(col("Numbers")).alias("Number"))
df_exploded.show()

+-------+------+
|   Name|Number|
+-------+------+
|  Alice|     1|
|  Alice|     2|
|  Alice|     3|
|    Bob|     4|
|    Bob|     5|
|    Bob|     6|
|Charlie|     7|
|Charlie|     8|
|Charlie|     9|
+-------+------+

number_2 = df_exploded.filter(col("Number") == 2)
number_2.show()

+----+------+
|Name|Number|
+----+------+
|Alice|     2|
+----+------+