ValueError:应为2D数组，而不是1D数组: array=[19. 27.896 0.1.0。]。使用array.reshape(-1，1)重塑数据

Question

我正在尝试使用K近邻来预测美国个人的保险成本

我已经成功地训练了数据集，当我尝试预测时，它开始处理错误

ValueError: Expected 2D array, got 1D array instead:
array=[19.    27.896  0.     1.     0.   ].
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.

同时执行以下代码：

import pandas as pd 
import numpy as np
from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors=1)


insurance = pd.read_csv(r"C:\Users\USER\Desktop\data set\insurance.csv")
insurance2 = pd.read_csv(r"C:\Users\USER\Desktop\data set\insurance.csv")


insurance['charges'] = insurance['charges'].astype('int')

sex_s=[]
for i in insurance['sex']:
    if i == "female":
        sex_s.append(1)
    elif i == "male":
        sex_s.append(0)
    
insurance2['sx']=sex_s

smk=[]
for y in insurance['smoker']:
    if y == "yes":
        smk.append(1)
    elif y == "no":
        smk.append(0)

insurance2['smks']=smk



insurance2 = insurance2.drop(['charges',"region", 'sex','smoker'], axis =1)

insurance2.rename(columns={'smks':'smoker', 'sx':'sex'}, inplace = True)


insurance2['bmi'] = insurance2['bmi'].astype('int')
insurance2['children'] = insurance2['children'].astype('int')
insurance2['sex'] = insurance2['sex'].astype('int')
insurance2['smoker'] = insurance2['smoker'].astype('int')
insurance2['age'] = insurance2['age'].astype('int')
insurance2.dtypes


X = insurance2
y = insurance['charges']


knn.fit(X,y)

knn.predict([19, 27.896, 0, 1, 0])

这是我第一次尝试预测

Answer 1

我认为你在最后一行漏掉了一对额外的方括号，如果你看一下文档(https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsClassifier.html)，你会看到他们用predict和下面的内容：

neigh.predict([[1.1]])

所以只要替换掉

knn.predict([19, 27.896, 0, 1, 0])

使用

knn.predict([[19, 27.896, 0, 1, 0]])

为了将来的参考，请确保首先查找您的问题以检查它是否存在，因为这是Error in Python script "Expected 2D array, got 1D array instead:"?的副本。