缩放和保留特征名称

Scailing and keeping names of features

我试图缩放数据框中的特征,但这也会导致我的 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)

输出数据: Output data

我之前的数据 Original Data

如果 X 具有所需的列名,则使用

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