动态列在java中怎么处理_java-如何在dataframe中动态地从列表中选择列和固定列

import org.apache.spark.sql.Dataset;

import org.apache.spark.sql.Row;

import org.apache.spark.sql.SparkSession;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import scala.collection.JavaConverters;

public class ATest {

public static void main(String[] args) {

SparkSession spark = SparkSession

.builder()

.appName("Java Spark SQL basic example")

.master("local[2]")

.getOrCreate();

List<String> cols = Arrays.asList("col_1", "col_2");

Dataset<Row> df = spark.sql("select 42 as ID, 'John' as NAME, 1 as col_1, 2 as col_2, 3 as col_3, 4 as col4");

df.show();

ArrayList<String> newCols = new ArrayList<>();

newCols.add("NAME");

newCols.addAll(cols);

df.select("ID", JavaConverters.asScalaIteratorConverter(newCols.iterator()).asScala().toSeq())

.show();

}

}