Python RandomForest, конвертирующий rfe.support_ в список

Question

После запуска модели RFE с RandomForests я получаю результат rfe.support_ (массив с True и False).

Если я сделаю df.columns [rfe.support_], я получу список сохраненных значений. Как я могу получить эти значения в виде списка? x.columns [rfe.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. Я хочу получить список сохраненных переменных, которые соответствуют 'true' в массиве rfe.support_.

Answer 1

Вы можете использовать функцию .tolist () как - list = rfe.support_.tolist ()

Теперь у вас есть список элементов массива

затем используйте цикл for, чтобы отфильтровать истинные значения из всего списка, созданного все время, при этом сохраняя отдельный список в качестве индекса истинных значений.

list2 = []

для i в диапазоне (1, len (список)): # итерация по списку

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

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