np.newaxis的作用

 

import numpy as np

x_data = np.linspace(0, 10, 10) 
print("x_data形状:", x_data.shape, "\nx_data值：", x_data)
x_data2 = x_data[:, np.newaxis]  # 将一维数组，转换为二维数组
print("\nx_data2形状:", x_data2.shape, "\nx_data2值：", x_data2)

结果（注意形状的改变）

x_data形状: (10,) 
x_data值： [ 0.          1.11111111  2.22222222  3.33333333  4.44444444  5.55555556
  6.66666667  7.77777778  8.88888889 10.        ]

x_data2形状: (10, 1) 
x_data2值： [[ 0.        ]
 [ 1.11111111]
 [ 2.22222222]
 [ 3.33333333]
 [ 4.44444444]
 [ 5.55555556]
 [ 6.66666667]
 [ 7.77777778]
 [ 8.88888889]
 [10.        ]]