如何在Java中使用Seq <String>从数据集中选择列?

时间:2019-07-29 09:49:11

标签: java apache-spark

我有一个Seq,需要用它来选择Java中的列

我知道函数{{1}},但没有第一列名称。

1 个答案:

答案 0 :(得分:1)

如果要使用Seq<String>进行选择,则可以拆分Seq来分别提取第一个元素:

Seq<String> columns = /* ... */;

Dataframe<Row> newDf = df.select(
    columns.apply(0), // first element
    columns.slice(1, columns.size()) // from the second to the end
);

也许首先检查columns的长度,以避免IndexOutOfBoundsException