在Scala中，如何对该组中的不同列进行分组并提取N个最高值？

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

val spark = SparkSession.builder()
  .appName("Group and Extract Top N Values")
  .master("local")
  .getOrCreate()

val data = Seq(
  ("A", 10),
  ("A", 20),
  ("A", 30),
  ("B", 15),
  ("B", 25),
  ("B", 35),
  ("C", 5),
  ("C", 15),
  ("C", 25)
).toDF("group", "value")

val topN = 2 // 提取前2个最高值

val result = data.groupBy("group")
  .agg(functions.sort_array(functions.collect_list("value")).as("sorted_values"))
  .select("group", functions.expr(s"sorted_values[${topN - 1}]").as("top_value"))

result.show()

+-----+---------+
|group|top_value|
+-----+---------+
|    B|       35|
|    A|       30|
|    C|       25|
+-----+---------+