如果键值满足Spark SQL2.4( <= 2.4版)中的条件，如何返回键值列表？)

import org.apache.spark.sql.functions._

val result = df.groupBy("key").agg(collect_list("value").alias("value_list"))

result.show(false)  // 打印结果

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

object KeyValueCollectionExample {
  def main(args: Array[String]): Unit = {
    // 创建 SparkSession
    val spark = SparkSession.builder()
      .appName("KeyValueCollectionExample")
      .getOrCreate()

    // 导入相关库
    import spark.implicits._
    import functions._

    // 创建示例数据
    val data = Seq(
      ("A", 1), ("A", 2), ("A", 3),
      ("B", 4), ("B", 5),
      ("C", 6)
    ).toDF("key", "value")

    // 使用 collect_list 函数返回键值列表
    val result = data.groupBy("key").agg(collect_list("value").alias("value_list"))

    // 打印结果
    result.show(false)

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

+---+----------+
|key|value_list|
+---+----------+
|A  |[1, 2, 3] |
|B  |[4, 5]    |
|C  |[6]       |
+---+----------+