作用
numpy.newaxis从字面上来理解就是用来创建新轴的,或者说是用来对array进行维度扩展的。
举例
>>> import numpy as np
>>> x = np.array([1, 2, 3, 4])
>>> print x.shape
>>> x1 = x[np.newaxis, :]
>>> print x1.shape
>>> print x1
>>> x2 = x[:, np.newaxis]
>>> print x2.shape
>>> print x2
实质上,我们以前用过None来对array进行维度扩展:
>>> y = x[None, :]
>>> print y.shape
>>> print y
numpy.newaxis的本质就是None:
>>> print np.axis
如果硬要说numpy.newaxis有什么作用的话,其实就是让读代码的人更明确这是在创建一个新轴罢了。