In [7]: np.linspace(1, 10, 10)
Out[7]: array([ 1., 2., 3., 4., 5., 6., 7., 8., 9., 10.])
In [8]: np.linspace(1, 10, 10).shape
Out[8]: (10,)
方案一:使用 np.newaxis
In [9]: np.linspace(1, 10, 10)[: , np.newaxis]
Out[9]:
array([[ 1.],
[ 2.],
[ 3.],
[ 4.],
[ 5.],
[ 6.],
[ 7.],
[ 8.],
[ 9.],
[10.]])
In [10]: np.linspace(1, 10, 10)[: , np.newaxis].shape
Out[10]: (10, 1)
方案二:使用None
In [11]: np.linspace(1, 10, 10)[: , None]
Out[11]:
array([[ 1.],
[ 2.],
[ 3.],
[ 4.],
[ 5.],
[ 6.],
[ 7.],
[ 8.],
[ 9.],
[10.]])
In [12]: np.linspace(1, 10, 10)[: , None].shape
Out[12]: (10, 1)
方案三:在reshape
函数中使用参数-1
参考:5个优雅的Numpy函数助你走出数据处理困境
In [13]: np.linspace(1, 10,10).reshape(-1, 1)
Out[13]:
array([[ 1.],
[ 2.],
[ 3.],
[ 4.],
[ 5.],
[ 6.],
[ 7.],
[ 8.],
[ 9.],
[10.]])
In [14]: np.linspace(1, 10,10).reshape(-1, 1).shape
Out[14]: (10, 1)