功能选择：尝试查看保留了哪些列时出现 AttributeError

Question

尝试将特征选择应用于我的分类 model 时出现错误。 我认为这个错误很简单，很容易解决，但我不确定如何解决。

这是我的代码：

X = df[['S_LENGTH', 'S_WIDTH', 'P_LENGTH', 'P_WIDTH']].values
y = df['SPECIES'].values

feature_select = SelectKBest(f_classif, k=20).fit(X, y)

但是当我尝试执行此操作以查看保留的功能时...

feature_select = X.columns[selection.get_support()]
print(features)

...我收到此错误：

AttributeError: 'numpy.ndarray' object has no attribute 'columns'

非常感谢有关如何解决此错误的任何帮助。

谢谢！

Answer 1

当您定义X时，您从 pandas DataFrame df中获取值。 这些值是numpy.ndarray ，因此没有属性columns 。

要修复，您需要更新您的行feature_select = X.columns[selection.get_support()] 。

它应该features = df.columns[feature_select.get_support()] 。