清除并保留功能名称

Question

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

输出数据：输出数据

我之前的数据原始数据

Answer 1

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

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