熊貓數據框中使用列名的測試數據選擇相同的列名

我正在尋找一種方法來自動改變功能位置train2.iloc[:,2]（代碼只給出了功能位置）功能名稱作爲我的測試數據需要這些相同的功能預測例如：熊貓數據框中使用列名的測試數據選擇相同的列名

columns = ['m2','m4','m12','m14'].

我的問題是，訓練列車數據後選擇的功能根據所使用的param_grid而變化。

我怎樣才能改變功能的位置功能在火車數據名稱，然後使用這些選擇在我的測試數據相同功能的名稱？

謝謝。

如果您知道位置（例如，在您的示例中爲2），則可以將其索引到列表列表中。即train_df.columns [2]，然後你可以使用你的測試集選擇列即test_df [train_df.columns [2]

2016-03-28 00:52:50 tvashtar

感謝您response..problem的是，我不知道哪些列已被選中，我只想要在列車數據中選擇的任何列也可以在測試數據中自動選擇。列車數據中選擇的列始終發生變化。 – isaacmoore7

你能舉一個例子嗎？還有一個返回值的例子？這聽起來像是你正在得到一個列索引列表，這使得很難理解實際問題。 – tvashtar

好的。 ..代碼使用GridSearchCV來確定要保留的重要功能 – isaacmoore7

回答