Как пропор способ разделить столбец в тестовом наборе на корзины?

Question

Я хочу разделить свой температурный столбец на ячейки, сейчас мой код выглядит следующим образом:

merged_data_train['temperature_categories']=pd.cut(merged_data_train.air_temperature,bins=20,labels=list(range(0,20)))
merged__data_test['temperature_categories']=pd.cut(merged__data_test.air_temperature,bins=20,labels=list(range(0,20)))

, но я не думаю, что это правильный способ применить его к каждому набору отдельно. Есть ли правильный способ сделать это преобразование? Может быть, что-то, что я мог бы вписать в состав поезда и применить на тестовом наборе. Спасибо

Answer 1

Вы ищете KBinsDiscretizer: https://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.KBinsDiscretizer.html