将数组的数组转换为pyspark中的结构的数组

from pyspark.sql import SparkSession
from pyspark.sql.types import ArrayType, StructType, StructField, StringType

spark = SparkSession.builder.getOrCreate()

array_of_arrays = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

array_schema = ArrayType(StructType([
    StructField("col1", StringType(), True),
    StructField("col2", StringType(), True),
    StructField("col3", StringType(), True)
]))

array_of_structs = spark.createDataFrame([(row,) for row in array_of_arrays], ["array_col"])
array_of_structs = array_of_structs.select(array_of_structs.array_col.cast(array_schema).alias("array_col"))