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