PySpark:如何分解两列数组

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

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

# 创建示例数据集
data = [(1, [1, 2, 3], [4, 5, 6]),
        (2, [7, 8], [9, 10]),
        (3, [11, 12, 13, 14], [15, 16, 17, 18])]

df = spark.createDataFrame(data, ["id", "array1", "array2"])

# 使用explode函数分解两列数组
df_exploded = df.select("id", explode("array1").alias("array1_element"), explode("array2").alias("array2_element"))

# 显示结果
df_exploded.show()