numpy.newaxis作用

总的来说newaxis是增加一个新的维度,

举例如下,a:1维数组,c和d都通过npy.newaxis变成二维数组,npy.newaxis在前面时就是增加列数,npy.newaxis在后面时增加了行数。

>> a=npy.array([1,2,3,4,5])
>>> c=a[npy.newaxis,:]
>>> c
array([[1, 2, 3, 4, 5]])
>>> a
array([1, 2, 3, 4, 5])

>>> d=a[:,npy.newaxis]
>>> d
array([[1],
       [2],
       [3],
       [4],
       [5]])
>>> a
array([1, 2, 3, 4, 5])

你可能感兴趣的:(numpy.newaxis作用)