从Spark数据框中删除多列

ras*_*iya 1 scala apache-spark

我正在尝试从数据框中删除多个列。我已经在/sf/answers/2787305181/上遵循了该示例,但是它对我不起作用。我正在使用Spark 2.4.0

这就是我现在正在做的事情。它确实有效,但是我想知道是否有更好的方法可以做到这一点。

val colsToRemove = Seq("colA", "colB", "colC", etc) 

var filterdDF = df
 for(i <- 0 until colsToRemove.size){
    filterdDF = filterdDF.drop(colsToRemove(i))
}
Run Code Online (Sandbox Code Playgroud)

Wil*_*ill 6

val filterdDF = df.drop(colsToRemove:_*)
Run Code Online (Sandbox Code Playgroud)