numpy.newaxis的使用

作用

        numpy.newaxis从字面上来理解就是用来创建新轴的，或者说是用来对array进行维度扩展的。

举例

>>> import numpy as np>>> x = np.array([1, 2, 3, 4])>>> print x.shape

• 1
• 2
• 3

>>> x1 = x[np.newaxis, :]>>> print x1.shape>>> print x1
1
2
3

>>> x2 = x[:, np.newaxis]>>> print x2.shape>>> print x2

• 1
• 2
• 3

        实质上，我们以前用过None来对array进行维度扩展：

>>> y = x[None, :]>>> print y.shape>>> print y
1
2
3

        numpy.newaxis的本质就是None：

>>> print np.axis

• 1

        如果硬要说numpy.newaxis有什么作用的话，其实就是让读代码的人更明确这是在创建一个新轴罢了。