获取与目标变量关联的特征列表

Question

我正在尝试获取大于0.15数据集的功能列表，它的代码允许我这样做，但结果是我有一系列的panda，我无法检索列表中的列名，以便在数据集中删除它们。我很感谢你的帮助。

# 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

Answer 1

它是Series，所以没有列。

对于所有按条件的值，请使用：

relevant_features = cor_target.index[cor_target>0.15]