scala：返回可变参数的方法

Question

从 scala 方法中，我想返回一个 Spark 列的变量编号，如下所示：

def getColumns() : (Column*) = {...}

这个想法是然后将它与spark sql一起使用：

myDf.select(getColumns, "anotherColumns"..)

问题是我有大约 30 个请求，它们都有相同的选择子句，我想将它们放在一起。

知道用...替换什么吗？我试过类似的东西：

($"col1", "$col2")

但它不能编译。

Answer 1

试试这个：

val df = Seq((1,2,3,4),(5,6,7,8)).toDF("a","b","c","d")

Typecast String 使用 map 函数来激发列，并根据需要在数组中附加附加列。

val lstCols = List("a","b")
df.select(lstCols.map(col) ++ List(col("c"),col("d")): _*).show()   

+---+---+---+---+
|  a|  b|  c|  d|
+---+---+---+---+
|  1|  2|  3|  4|
|  5|  6|  7|  8|
+---+---+---+---+