1. 程式人生 > >numpy.linspace和numpy.newaxis

numpy.linspace和numpy.newaxis

>>> import numpy as np

numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None)

在指定的間隔內返回均勻間隔的數字。在[start, stop]這個區間的端點可以任意的被排除在外,預設包含端點;retstep=True時,顯示間隔長度。

>>> np.linspace(2.0,3.0,num=5)
array([2.  , 2.25, 2.5 , 2.75, 3.  ])
>>> np.linspace(2.0,3.0,num=5,endpoint=False)
array([2. , 2.2, 2.4, 2.6, 2.8])
>>> np.linspace(2.0,3.0,num=5,retstep=True)
(array([2.  , 2.25, 2.5 , 2.75, 3.  ]), 0.25)

numpy.newaxis None的方便別名,對於索引陣列是有用的。

>>> np.newaxis is None
True
>>> x = np.array([1,2,3]) # 一維陣列[1,2,3]
>>> x
array([1, 2, 3])
>>> x.shape
(3,)
>>> x[:,np.newaxis]# 二維陣列
array([[1],
       [2],
       [3]])
>>> x[:,np.newaxis].shape
(3, 1)
>>> x[np.newaxis,:]# 二維陣列
array([[1, 2, 3]])
>>> x[np.newaxis,:].shape
(1, 3)
>>> x[:,np.newaxis,np.newaxis]# 三維陣列
array([[[1]],

       [[2]],

       [[3]]])
>>> x[:,np.newaxis,np.newaxis].shape
(3, 1, 1)