numpy中的np.newaxis

import numpy as np

x_data=np.linspace(-0.5,0.5,10)
print(x_data)

输出结果为
[-0.5        -0.38888889 -0.27777778 -0.16666667 -0.05555556  0.05555556
  0.16666667  0.27777778  0.38888889  0.5       ]
十个浮点数，一维度
1行

---

x_data=np.linspace(-0.5,0.5,10)[:,np.newaxis]
print(x_data)

输出结果为
[[-0.5       ]
 [-0.38888889]
 [-0.27777778]
 [-0.16666667]
 [-0.05555556]
 [ 0.05555556]
 [ 0.16666667]
 [ 0.27777778]
 [ 0.38888889]
 [ 0.5       ]]
10行1列，二维度

这样看来，[:,np.newaxis]说白了就是添加一个维度，我们再来试一下

import numpy as np

x=np.ones(3)
print(x)

x=np.ones(3)[:,np.newaxis]
print(x)

x=np.ones(3)[np.newaxis,:]
print(x)

输出结果为

[1. 1. 1.]
[[1.]
 [1.]
 [1.]]
[[1. 1. 1.]]

意思是np.newaxis加在后边就是加上列，加在前边就是再套一行