如何使用平面映射在spark scala dataframe列上执行一对多映射

import org.apache.spark.sql.{DataFrame, SparkSession}
import org.apache.spark.sql.functions._

val spark = SparkSession.builder()
  .appName("DataFrame One-to-Many Mapping")
  .getOrCreate()

val data = Seq(("A", "1,2,3"), ("B", "4,5"), ("C", "6"))
val df = spark.createDataFrame(data).toDF("col1", "col2")

def oneToManyMapping: String => Seq[String] = (input: String) => {
  input.split(",").toSeq
}

val mappedDF = df.withColumn("mapped_col", explode(split(col("col2"), ",")))

mappedDF.show()