Scala:将dataframe列复制到数组中，并保留原始顺序

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

object Main {
  def main(args: Array[String]): Unit = {
    // 创建SparkSession
    val spark = SparkSession.builder()
      .appName("Copy DataFrame Column to Array")
      .master("local[*]")
      .getOrCreate()

    // 创建示例DataFrame
    val df = spark.createDataFrame(Seq(
      (1, "John"),
      (2, "Jane"),
      (3, "Alice")
    )).toDF("id", "name")

    // 将DataFrame列复制到数组中并保留原始顺序
    val columnArray = df.select("name").collect().map(_.getString(0))

    // 打印数组内容
    columnArray.foreach(println)

    // 停止SparkSession
    spark.stop()
  }
}