Python RandomForest将rfe.support_转换为list

Question

在使用RandomForests运行RFE模型之后，我得到了rfe.support_结果(具有True和False的数组)。

如果我执行df.columnsrfe.support_，我会得到一个保留值的列表。如何以列表形式获取此值？x.columnsrfe.support_似乎是一个不可调用的对象。

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_

rfe.support_是一个包含false和true的数组，我想要的是获得与rfe.support_数组中的'true‘相对应的保留变量的列表。

Answer 1

您可以使用to '.tolist()‘函数作为- list = rfe.support_.tolist()

现在，您已经有了数组项的列表

然后使用for循环从始终创建的整个列表中过滤出true值，同时维护一个单独的列表作为true值的索引。

list2 = []

for i in range(1，len( list ))：#迭代列表

if(int(list[i])==1):          # for the true values

    list2.append(i)           # the separate list with the index of true value items