将多个列作为Seq/Array传递给Scala Spark中的UDF

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

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

// 导入隐式转换，以便使用$符号访问列
import spark.implicits._

// 创建示例数据
val data = Seq(
  ("Alice", 25, 100),
  ("Bob", 30, 200),
  ("Charlie", 35, 300)
).toDF("name", "age", "salary")

// 定义一个函数，将多个列相加
def sumColumns(cols: Seq[Int]): Int = {
  cols.sum
}

// 将函数转换为UDF
val sumColumnsUDF = udf(sumColumns _)

// 使用UDF计算新列
val result = data.withColumn("sum", sumColumnsUDF(array($"age", $"salary")))

// 显示结果
result.show()