Pandas dataframe использует имена столбцов в данных поезда для выбора одинаковых имен столбцов в данных теста

Question

Я ищу способ автоматического изменения позиции функции train2.iloc[:,2] (код только дает положение функции) для имени функции, так как мои тестовые данные нуждаются в этих же функциях для прогнозирования, например:

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

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

Как изменить положение функции на имя функции в данных поезда, а затем использовать их для выбора тех же имен функций в моих тестовых данных?

спасибо.

Answer 1

Если вы знаете позицию (т. Е. 2 ​​в вашем примере), вы можете просто индексировать ее в списке столбцов. т.е. train_df.columns [2], вы могли бы использовать это, чтобы выбрать столбец в тестовом наборе, т. е. test_df [train_df.columns [2]]