numpy.newaxis的使用

作用

        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有什么作用的话,其实就是让读代码的人更明确这是在创建一个新轴罢了。





猜你喜欢

转载自blog.csdn.net/u013745804/article/details/79634196