我正在寻找一种方法来自动更改功能位置train2.iloc[:,2]（代码仅提供功能位置）到功能名称，因为我的测试数据需要这些相同的功能进行预测，例如：
columns = ['m2','m4','m12','m14'].
我的问题是训练列车数据后选择的功能会根据所使用的param_grid而改变。
如何在列车数据中将要素位置更改为要素名称，然后使用它们在我的测试数据中选择相同的要素名称？
谢谢。
答案 0 :(得分：0)
如果你知道位置（例如你的例子中的2），你可以直接索引列列表。即train_df.columns [2]，然后您可以使用它选择测试集中的列，即test_df [train_df.columns [2]]