如何在变换不是1:1而是1:多的情况下创建火花DataSet

import org.apache.spark.sql.{SparkSession, Row}
import org.apache.spark.sql.types.{StructType, StructField, StringType}

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Create Spark DataSet")
  .master("local")
  .getOrCreate()

// 创建输入数据集
val input = spark.sparkContext.parallelize(Seq("1,2,3", "4,5", "6,7,8,9"))

// 定义转换函数
def transform(line: String): Seq[Row] = {
  val values = line.split(",")
  values.map(value => Row(value))
}

// 定义输出数据集的Schema
val schema = StructType(Seq(StructField("value", StringType, nullable = true)))

// 应用转换函数并创建火花DataSet
val output = spark.createDataFrame(input.flatMap(transform), schema)

// 打印结果
output.show()