ValueError: Expected 2D array, got 1D array instead: 的解决办法

  1. 在学knn最邻近分类时,原始数据根据参数Q1、Q2分为type=T1、type=T2两种;

  2. 执行

knn = neighbors.KNeighborsClassifier()
knn.fit(df[['Q1','Q2']],df['type']) 
knn.predict([35,26])
  1. 报错

ValueError: Expected 2D array, got 1D array instead:
array=[35 26].
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.

  1. 在尝试了
np.array([35,26]).reshape(1,-1)

结果为
结果
之后,了解到把一维数组改为二维数组只要外层加一个[]就可以了,可以省去reshape。

  1. 最终代码
knn = neighbors.KNeighborsClassifier()   
knn.fit(df[['Q1','Q2']],df['type'])   
knn.predict([[35,26]])
  1. 运行结果
    在这里插入图片描述

你可能感兴趣的:(python)