如何在Scala - Apache Spark中查找数组中值的平均值

import org.apache.spark.sql.SparkSession

val spark = SparkSession.builder().appName("ArrayAverage").master("local").getOrCreate()

val arrayRDD = spark.sparkContext.parallelize(Array(1, 2, 3, 4, 5))

val sum = arrayRDD.reduce(_ + _)

val count = arrayRDD.count()

val average = sum.toDouble / count.toDouble

println("数组的平均值为：" + average)

import org.apache.spark.sql.SparkSession

object ArrayAverage {
  def main(args: Array[String]): Unit = {
    val spark = SparkSession.builder().appName("ArrayAverage").master("local").getOrCreate()
    val arrayRDD = spark.sparkContext.parallelize(Array(1, 2, 3, 4, 5))
    val sum = arrayRDD.reduce(_ + _)
    val count = arrayRDD.count()
    val average = sum.toDouble / count.toDouble
    println("数组的平均值为：" + average)
    spark.stop()
  }
}