如何将Boruta筛选的特征转换为带列名的Pandas DataFrame

直接利用Pandas的DataFrame构造函数即可完成转换，具体操作如下：

步骤说明 #

1. 先确保已导入pandas库：

import pandas as pd

2. 准备好两个核心变量：

   • selected_features：Boruta筛选后得到的特征列名集合/列表（若为集合需转成有序列表）
   • X_filtered：numpy数组格式的筛选后特征数据

3. 执行转换：

# 若selected_features是集合类型，先转为列表保证列顺序
if isinstance(selected_features, set):
    selected_features = list(selected_features)

# 生成带列名的DataFrame
X_filtered_df = pd.DataFrame(X_filtered, columns=selected_features)

关键注意点 #

• 必须保证X_filtered的列数与selected_features的元素数量完全一致，否则会抛出维度不匹配的错误
• 集合类型是无序的，转为列表后能确保列名顺序与数组中特征的顺序对应

内容的提问来源于stack exchange，提问作者user7446890