基于Scala中的另一列聚合来自dataframe列的数据

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

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("AggregationExample")
  .getOrCreate()

// 导入隐式转换，以便使用DataFrame的操作方法
import spark.implicits._

// 创建DataFrame
val df = Seq(
  ("A", 1),
  ("A", 2),
  ("B", 3),
  ("B", 4),
  ("C", 5)
).toDF("col1", "col2")

// 使用groupBy和agg进行聚合
val result = df.groupBy("col1")
  .agg(functions.sum("col2").alias("sum_col2"))

// 显示结果
result.show()