Получить список функций, связанных с целевой переменной - PullRequest
Новая
       14

Получить список функций, связанных с целевой переменной

1 голос
/ 03 мая 2020

Я пытаюсь получить список возможностей моего набора данных больше 0,15. Этот код позволяет мне это сделать, но в результате у меня есть панда ser ie, которой я не могу получить имена столбцов в списке, чтобы после удаления их в моем наборе данных. Я ценю вашу помощь. 

# Remove highly correlated features
cor = features_binario.corr()
#Correlation with output variable
cor_target = abs(cor["G3"])

#Selecting highly correlated features
relevant_features = cor_target[cor_target>0.15]

first_column = relevant_features.iloc[:,0]
first_column

Я получаю эту ошибку, IndexingError: Слишком много индексаторов 

Var: relevant_features
school_GP      0.177564
school_MS      0.177564
higher_no      0.251587
higher_yes     0.251587
course_math    0.168394
course_por     0.168394
age            0.153819
Medu           0.184047
Fedu           0.183627
failures       0.390165
G1             0.701693
G2             0.717439
G3             1.000000
Name: G3, dtype: float64

1 Ответ

2 голосов
/ 03 мая 2020

Это Series, поэтому столбцы отсутствуют.

Для всех значений по условиям используйте: 

relevant_features = cor_target.index[cor_target>0.15]
...