python - 熊猫在缩放后返回所有列

Question

我有一个包含 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。我如何实现这一目标？

标签： pythonpandas

Answer 1

您可以使用numpy 连接。

np.concatenate((x_train_new, x_train['C']), axis=1)