numpy重塑矩阵

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)