如何使用List<String>选择dataset的列

import java.util.ArrayList;
import java.util.List;

public class DatasetColumnSelection {
    public static void main(String[] args) {
        // 创建一个List对象，存储要选择的列名
        List<String> selectedColumns = new ArrayList<>();
        selectedColumns.add("column1");
        selectedColumns.add("column3");

        // 假设dataset是一个包含多个列的数据集合
        List<String> datasetColumn1 = new ArrayList<>();
        datasetColumn1.add("data11");
        datasetColumn1.add("data12");
        datasetColumn1.add("data13");

        List<String> datasetColumn2 = new ArrayList<>();
        datasetColumn2.add("data21");
        datasetColumn2.add("data22");
        datasetColumn2.add("data23");

        List<String> datasetColumn3 = new ArrayList<>();
        datasetColumn3.add("data31");
        datasetColumn3.add("data32");
        datasetColumn3.add("data33");

        // 创建一个新的List，存储选择的列数据
        List<String> selectedData = new ArrayList<>();

        // 遍历selectedColumns，选择对应的列数据
        for (String columnName : selectedColumns) {
            if (columnName.equals("column1")) {
                selectedData.addAll(datasetColumn1);
            } else if (columnName.equals("column2")) {
                selectedData.addAll(datasetColumn2);
            } else if (columnName.equals("column3")) {
                selectedData.addAll(datasetColumn3);
            }
        }

        // 输出选择的列数据
        System.out.println("Selected Data:");
        for (String data : selectedData) {
            System.out.println(data);
        }
    }
}