如何从一列数组中提取元素，并将这些元素存储在pyspark中的新数据帧中？

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

spark = SparkSession.builder.getOrCreate()

data = [("A", [1, 2, 3]), ("B", [4, 5, 6]), ("C", [7, 8, 9])]
df = spark.createDataFrame(data, ["ID", "ArrayColumn"])

exploded_df = df.select(col("ID"), explode(col("ArrayColumn")).alias("Element"))

new_df = exploded_df.groupBy("ID").agg(collect_list("Element").alias("NewArray"))

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

spark = SparkSession.builder.getOrCreate()

data = [("A", [1, 2, 3]), ("B", [4, 5, 6]), ("C", [7, 8, 9])]
df = spark.createDataFrame(data, ["ID", "ArrayColumn"])

exploded_df = df.select(col("ID"), explode(col("ArrayColumn")).alias("Element"))

new_df = exploded_df.groupBy("ID").agg(collect_list("Element").alias("NewArray"))

new_df.show()