有没有从SelectFromModel方法输出选定的列名？

Question

我在加载为DataFrame的数据集中使用ExtraTreesClassifier和SelectFromModel执行功能选择，但是我想将这些选定的功能作为DataFrame保存到csv文件，同时保持列名称。 请注意，输出是numpy数组返回重要功能整列而不是列标题

 try {
        String line;
        Process p = Runtime.getRuntime().exec(System.getenv("windir") + "\\system32\\" + "tasklist.exe");
        BufferedReader input
                = new BufferedReader(new InputStreamReader(p.getInputStream()));
        while ((line = input.readLine()) != null) {
            System.out.println(line); //<-- Parse data here.
        }

        input.close();
    } catch (Exception err) {
        err.printStackTrace();
    }

Answer 1

model = SelectFromModel(clf, prefit=True)
feature_idx = model.get_support()
feature_name = df.columns[feature_idx]

Answer 2

使用DataFrame.to_csv()方法将数据框保存为csv文件。

执行以下操作：

X_new.to_csv("your/path", sep=';')

Here是该方法文档的链接。