是否有從SelectFromModel方法輸出選定的列名稱？

Question

我使用ExtraTreesClassifier和SelectFromModel在加載爲DataFrame的數據集中執行了特徵選擇，但是我想將這些選定的特徵作爲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) 

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是該方法的文檔的鏈接。