Pyspark将列表转换为特定列中的字典

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, create_map, lit

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

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

# 创建DataFrame
df = spark.createDataFrame(data, ["name", "fruits"])

# 将列表转换为字典
df = df.withColumn("fruits_dict", create_map(*[lit(x) for x in col("fruits")]))

# 显示结果
df.show(truncate=False)