import numpy as np
import pandas as pd
from sklearn.feature_selection import RFE
from sklearn.linear_model import LinearRegression
def excel_one_line_to_list():
X = pd.read_excel("xxx.xls", usecols=[0,1,2,3,4,5],names=None) # 读取项目名称列,不要列名
names = X.columns.tolist()
X = np.array(X)
X = X.tolist()
Y = pd.read_excel("xxx.xls", usecols=[6],names=None) # 读取项目名称列,不要列名
Y = np.array(Y)
Y = Y.tolist()
#use linear regression as the model
lr = LinearRegression()
#rank all features, i.e continue the elimination until the last one
rfe = RFE(lr, n_features_to_select=7) #n_features_to_select是想要选择的特征个数
rfe.fit(X,Y)
print("Features sorted by their rank:")
print(sorted(zip(rfe.ranking_, names)))
if __name__ == '__main__':
excel_one_line_to_list()