Масштабирование и сохранение имен функций

Я пытался масштабировать функции в своем фрейме данных, однако это также приводит к потере всех имен в моем df. Код:

from sklearn.preprocessing import RobustScaler
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42)
scaler = RobustScaler().fit(X_train)
X_train = scaler.transform(X_train)
pd.DataFrame(X_train)

Выходные данные: Выходные данные

Данные, которые у меня были до этого Исходные данные

0
0
15
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

если X имеет желаемые имена столбцов, используйте

pd.DataFrame(X_train, columns=X.columns)

Другие вопросы по теме