在Scala Spark中，如何分组并将组中的每个值除以该组中的行数

import org.apache.spark.sql.SparkSession

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

    // 创建示例数据
    val data = Seq(
      ("group1", 1),
      ("group1", 2),
      ("group1", 3),
      ("group2", 4),
      ("group2", 5)
    )

    // 将数据转换为DataFrame
    import spark.implicits._
    val df = data.toDF("group", "value")

    // 分组并计算每个组的行数
    val grouped = df.groupBy("group").count()

    // 将组中的每个值除以该组中的行数
    val result = df.join(grouped, Seq("group"))
      .select($"group", $"value" / $"count")

    // 显示结果
    result.show()

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