小白还是记录一下今天踩的坑吧
见如下代码:
x = np.array([1, 2, 3, 4, 5]) y = np.array([[1],[2],[3],[4],[5]]) print(x.shape) print(y.shape)
输出为
(5,) (5, 1)
两者的区别显而易见:上面的数是一维数组,下面的数是二维数组。
两者之间的转换:
二维数组转换为一维数组:
y = y.flatten() print(y) print(y.shape)
输出的结果为
[1 2 3 4 5] (5,)
可以看到,二维数组就这样轻松愉快的转为一维数组了



