无法使用“ from sklearn.preprocessing import Imputer”,它显示了异常“数据必须为一维”

时间:2019-07-26 05:55:59

标签: python-3.x pandas numpy

我为人工神经网络(ANN)建立了模型。我想在训练模型之前对数据进行预处理。 我已经尝试了下面给出的代码。

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

dataset = pd.read_csv('Update-Detaset with hacking1.csv')
y=[]
X = dataset.iloc[:,2:7]
y = dataset.iloc[:,8]
from sklearn.preprocessing import Imputer
imputer = Imputer(missing_values = 'NaN', strategy = 'mean', axis = 0)
Y = np.reshape(y,(-1,1))
imputer = imputer.fit(Y)
Y= imputer.transform(Y)

Exception: Data must be 1-dimensional

在这里,Update-Detaset with hacking1.csv是.csv文件。数据集如下所示: Please click the link to see the demo of the csv file  它显示以下错误。 我该如何解决?

1 个答案:

答案 0 :(得分:0)

这与Imputer无关。您应该能够从引发异常的行号中看出这一点。错误是由于您试图重塑熊猫DataFrame而引起的。更改

y = dataset.iloc[:,8]

y = dataset.iloc[:,8].values

它应该可以工作。