np.newaxis的使用方法

最近在学习机器学习的时候，学习到一个为一维数组添加维度的函数np.newaxis

import numpy as np
x = np.arange(10)
x

这是一个普通的一维数组，输出如下：

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

下面我们为其添加维度，有两种方式：

一、使每个元素作为一维

X = x[:, np.newaxis]
X

输出：

array([[0],
       [1],
       [2],
       [3],
       [4],
       [5],
       [6],
       [7],
       [8],
       [9]])

二、将整个数组作为一维

x = x[np.newaxis, :]
x

输出：

array([[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]])

后续有更深理解了继续更