当前位置: 代码迷 >> python >> 熊猫在缩放后返回所有列
  详细解决方案

熊猫在缩放后返回所有列

热度:63   发布时间:2023-07-16 11:10:50.0

我有一个包含 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 的所有四列,就像它一样。 我如何实现这一目标?

您可以使用 。

np.concatenate((x_train_new, x_train['C']), axis=1)
  相关解决方案