在使用randomforest运行rfe模型之后,我得到一个rfe.support结果(带有true和false的数组)。
如果我做了df.columns[rfe.support\u]我会得到一个保留值列表。如何将此值作为列表获取?x.columns[rfe.support\u]似乎是不可调用的对象。
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris
iris = load_iris()
rfe = RFE(estimator=clf_rf_3, n_features_to_select=2, step=1)
rfe = rfe.fit(iris["data"], iris["target"])
rfe.support_
support是一个具有false和true的数组,我想得到一个与rfe.support数组中的“true”相对应的保留变量列表。