val columnName=Seq("col1","col2",....."coln");
有没有办法做 dataframe.select 操作来获取只包含指定列名的数据帧。
我知道我可以做
dataframe.select("col1","col2"...)
但columnName
在运行时生成。我可以做
dataframe.select()
对循环中的每个列名重复。它是否有任何性能开销?。有没有其他更简单的方法来实现这一点?
最佳答案
val columnNames = Seq("col1","col2",....."coln")
// using the string column names:
val result = dataframe.select(columnNames.head, columnNames.tail: _*)
// or, equivalently, using Column objects:
val result = dataframe.select(columnNames.map(c => col(c)): _*)
关于Scala Spark DataFrame : dataFrame. 根据列名序列选择多个列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36131716/