问题描述
我有一个包含 A、B、C、D 列的数据框
A、B 是数字列 C、D 是非数字列
我只想缩放 A 列和 B 列
这是我的代码
columnsToScale = ['A','B']
x_train, x_test, y_train, y_test = train_test_split(x, y, random_state=0)
scaler = MinMaxScaler()
x_train_new = scaler.fit_transform(x_train[columnsToScale])
执行此操作后,我的x_train_new
只有 A 列和 B 列。我希望它具有 A 和 B 缩放以及 C、D 的所有四列,就像它一样。
我如何实现这一目标?
1楼
您可以使用 。
np.concatenate((x_train_new, x_train['C']), axis=1)