关于np.newaxis的理解

经常在sklearn上看到np.newaxis，这里记录一下我的理解

np.arange(0, 10)

这句话 生成的是一个一维的数组，如下：

[0 1 2 3 4 5 6 7 8 9]

输出其shape：(10,)

那么我如何才能将其转化为shape=(10,1)呢

可以用两种方法：

1.使用shape

y=np.arange(1, 11)
y.shape=(10,1)
print(y)

1.  

结果如下：

 [[ 1]
[ 2]
[ 3]
[ 4]
[ 5]
[ 6]
[ 7]
[ 8]
[ 9]
[10]]

1.  

2. 使用np.newaxis

 

print(np.arange(0, 10)[:, np.newaxis])

结果如下：

 

 [[0]
[1]
[2]
[3]
[4]
[5]
[6]
[7]
[8]
[9]]

1.  

如上所示，应该清楚了np.newaxis了吧

上面的代码实质就是将原本的(10,)移到行上，然后新增一列