sklearn报错：Expected 2D array, got 1D array instead

背景

使用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),大功告成！

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容