featureSelector = SelectKBest(score_func=f_regression, k=2)
featureSelector.fit(X_train_data, Y_train_data)
print([(1 + zero_based_index) for zero_based_index in list(featureSelector.
    get_support(indices=True))])
Scikit-learn feature selection for regression data