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