# correlation corr = train.corr() # sort in descending order corr_top = corr['SalePrice'].sort_values(ascending=False)[:10] top_features = corr_top.index[1:] corr_top