创建一个spark dataframe列，它包含一个列表作为数据类型

from pyspark.sql import SparkSession
from pyspark.sql.types import ArrayType, StringType
from pyspark.sql.functions import col, lit

spark = SparkSession.builder.getOrCreate()

data = [["John", ["apple", "banana", "orange"]],
        ["Alice", ["grape", "melon", "strawberry"]],
        ["Bob", ["pineapple", "watermelon", "kiwi"]]]

schema = ["name", "fruits"]

df = spark.createDataFrame(data, schema)

df.show()

+-----+------------------+
| name|            fruits|
+-----+------------------+
| John|[apple, banana, orange]|
|Alice|[grape, melon, strawberry]|
|  Bob|[pineapple, watermelon, kiwi]|
+-----+------------------+