np.newaxis的一点理解

import numpy as np
print(type(np.newaxis))

首先先看newaxis的类型

结果：<class 'NoneType'>

也就是newaxis == None

import numpy as np
x=np.arange(4)
print(x)
x.shape=(4,1)
print(x)

import numpy as np
x=np.arange(4)
print(x)
print(x[:,np.newaxis])

结果：

[0 1 2 3]
[[0]
 [1]
 [2]
 [3]]

这上面两种方法得到的结果都是一样的，他们两者有什么区别呢？

首先第一种方法只能输入x.shape=(4,1)或者x.shape=(1,4)

比如你要是想改成 x.shape=(4,2）

就会得到报错：ValueError:cannot reshape array of size 4 into (4,2)

但是第二种方法

当我们输入 print(x[1:,np.newaxis])时，就可以得到

[[1]

[2]

[3]]

 

参考:https://blog.csdn.net/molu_chase/article/details/78619731