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

Question

我有一个 Seq，需要用它来选择 Java 中的列我知道该函数.select(String col, Seq<String> cols)，但我没有第一列名称。

Answer 1

如果你想使用 a 选择，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