斯克莱恩的熊猫错误：1列熊猫的错误

Question

我正在对熊猫的一列应用缩放，但它给了我一个错误

import pandas as pd 
import numpy as np
from sklearn.preprocessing import MinMaxScaler, StandardScaler
df:
group people value value_50
  1    5      100    1
  2    2      90     1
  1    10     80     1
  2    20     40     0
  1    7      10     0
  2    23     30     0
df = pd.read_clipboard()

选择一列

y_train = df.value
sc = StandardScaler()
y_train_sc = sc.fit_transform(y_train)
y_train_sc = sc.fit_transform(y_train.values)

但是两个尝试都出现类型错误

 ValueError: Expected 2D array, got 1D array instead:

如何用熊猫和 sklearn 来做？

Answer 1

使用

y_train_sc = sc.fit_transform(y_train.to_frame())

或选择作为y_train= df[['value']] 。当StandardScaler需要类似 2D 数据帧的输入时，选择该列作为 1D 系列。