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),大功告成!

上一篇下一篇

猜你喜欢

热点阅读