sklearn报错:Expected 2D array, got
2020-03-30 本文已影响0人
脸白涂的蜡
背景
使用sklearn时会报错
ValueError: Expected 2D array, got 1D array instead
原因
这是因为在新版的sklearn中要求所有的数据都应该是二维矩阵,所以当数据是单独的一行或者一列时,要将其修改成二维。
解决方法
使用.reshape(1,-1)即可
x = [1,2,3]
x_new = np.array(x).reshape(1,-1)
print(x_new.shape)
输出结果为(1,3),大功告成!