与spark map列中的最大值对应的键

import org.apache.spark.sql.SparkSession

object SparkMaxKey {
  def main(args: Array[String]): Unit = {
    val spark = SparkSession.builder()
      .appName("SparkMaxKey")
      .master("local")
      .getOrCreate()

    // 创建键值对数据集
    val data = Seq(("key1", 10), ("key2", 20), ("key3", 30), ("key4", 20))
    val rdd = spark.sparkContext.parallelize(data)

    // 将键值对数据集转换为只包含值的数据集
    val values = rdd.map(_._2)

    // 找到最大值
    val maxValue = values.max()

    // 筛选出与最大值相等的值
    val filteredValues = values.filter(_ == maxValue)

    // 与原始键值对数据集进行连接，获取与最大值对应的键
    val result = rdd.filter { case (_, value) => filteredValues.contains(value) }

    result.foreach(println)

    spark.stop()
  }
}