Scala Spark-> 从 DataFrame 中选择前 15 列

Posted 2023-04-15

【中文标题】Scala Spark-> 从 DataFrame 中选择前 15 列

【英文标题】：Scala Spark-> Select first 15 columns from a DataFrame

【发布时间】：2020-11-26 14:37:21

【问题描述】：

我正在尝试从包含 500 多个列的 DataFrame 中获取前 15 列。但我不知道该怎么做，因为我是第一次使用 Scala Spark。

我正在搜索但没有找到任何东西，只是如何按名称获取 cols，例如：

val df2 = df.select("firstColName", "secondColeName")

我怎样才能通过索引做到这一点？

提前致谢！

【参考方案1】：

Scala 示例：

df.selectExpr(df.columns.take(15):_*)

【讨论】：

这就是我要找的！非常感谢：D

df.columns 的类型为 Array[String]，那么您需要将其映射到 Array[Column]： df.select(df.columns.take(15).map(col) :_ *)