【np.newaxis用法】增加维度

脑子

1. >>> np.linspace(1,-1,5)
   
2. array([ 1. ,  0.5,  0. , -0.5, -1. ])
   
3. #=============================================================
   
4. >>> np.linspace(1,-1,5)[:,np.newaxis]
   
5. array([[ 1. ],
   
6.        [ 0.5],
   
7.        [ 0. ],
   
8.        [-0.5],
   
9.        [-1. ]])
   
10. >>> np.linspace(1,-1,5)[:,np.newaxis].shape
    
11. (5, 1)
    
12. #==============================================================
    
13. >>> np.linspace(1,-1,5)[np.newaxis,:]
    
14. array([[ 1. ,  0.5,  0. , -0.5, -1. ]])
    
15. >>> np.linspace(1,-1,5)[np.newaxis,:].shape
    
16. (1, 5)

复制代码

_2_

看着很乱……
改一下代码段吧