平均像素的scala返回矩阵

import org.apache.spark.ml.linalg.{Matrix, Vectors}
import org.apache.spark.ml.stat.Summarizer
import org.apache.spark.sql.SparkSession

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

    // 创建示例数据
    val data = Seq(
      Vectors.dense(1.0, 2.0, 3.0),
      Vectors.dense(4.0, 5.0, 6.0),
      Vectors.dense(7.0, 8.0, 9.0)
    )

    // 将数据转换为DataFrame
    import spark.implicits._
    val df = data.map(Tuple1.apply).toDF("features")

    // 使用Summarizer计算平均像素
    val result = df.select(Summarizer.mean($"features").as("average"))

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

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