python - 有没有办法从SelectFromModel方法输出选定的列名称？

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

import pandas as pd
from sklearn.ensemble import ExtraTreesClassifier
from sklearn.feature_selection import SelectFromModel
import numpy as np


df = pd.read_csv('los_10_one_encoder.csv')
y = df['LOS'] # target
X= df.drop('LOS',axis=1) # drop LOS column
clf = ExtraTreesClassifier()
clf = clf.fit(X, y)
print clf.feature_importances_

model = SelectFromModel(clf, prefit=True)
X_new = model.transform(X)

最佳答案

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

关于python - 有没有办法从SelectFromModel方法输出选定的列名称？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/41088576/