np.newaxis顾名思义,增加新的维度,使用方法如下:
X = np.linspace(0,6,6)[:, np.newaxis]
y = np.linspace(0,6,6)[np.newaxis, :]
print('X:', X)
print('y:', y)
输出结果如下:
X: [[ 0. ]
[ 1.2]
[ 2.4]
[ 3.6]
[ 4.8]
[ 6. ]]
y: [[ 0. 1.2 2.4 3.6 4.8 6. ]]
所以就是冒号保持现有结构,并增加np.newaxis所在的维度
注:np.linspace(0,6,6)就是在0和6之间均匀的选择6个数字,返回ndarray
仅供学习参考,欢迎指正