如何获取PyCaret建模后生成的数据集列名？

你可以通过PyCaret内置的get_config()函数获取处理后的数据集列名，以下是具体实现方法：

• 获取特征列名（不含目标列）
  执行setup后，处理后的特征训练集被存储在X_train配置项中，直接提取其列名即可：

  # 获取处理后的特征训练集
  processed_features = get_config('X_train')
  # 输出列名列表
  print(processed_features.columns.tolist())
  

• 获取包含目标列的完整数据集列名
  如果你需要包含目标列charges的全部列名，可以提取完整训练集的列名：

  # 获取处理后的完整训练集（含目标列）
  full_train_data = get_config('train_data')
  # 输出所有列名
  print(full_train_data.columns.tolist())
  

补充说明：setup生成的训练集和测试集列名完全一致，因此也可以通过get_config('X_test')或get_config('test_data')获取列名，结果完全相同。

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