Python中range、np.arange和np.linspace的區(qū)別
1. range
range是python內(nèi)置的一個類,該類型表示一個不可改變(immutable)的數(shù)字序列,常常用于在for循環(huán)中迭代一組特殊的數(shù),它的原型可以近似表示如下:
class range(stop) class range(start, stop, step=1)
(注意,Python是不允許定義兩個類初始化函數(shù)的,其實(shí)其CPython實(shí)現(xiàn)更像是傳入不定長參數(shù)*args,然后根據(jù)len(args)來進(jìn)行不同的拆分,但我們這里遵循Python文檔風(fēng)格寫法)
如果只傳入stop參數(shù),那么我們就默認(rèn)在[0, stop)區(qū)間以步長1進(jìn)行迭代。如果傳入2或3個參數(shù),則我們會將在[start, stop)區(qū)間以step步長(可選,默認(rèn)為1)迭代 。注意,三個參數(shù)必須全部為整數(shù)值。
它的常見使用樣例如下:
print(list(range(10))) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] print(list(range(0, 30, 5))) # [0, 5, 10, 15, 20, 25]
當(dāng)stop<=start時,而直接采用默認(rèn)的step=1時,元素會為空:
print(list(range(0))) # [] print(list(range(1, 0))) # []
此時的迭代我們需要將迭代步長設(shè)置為負(fù):
print(list(range(0, -10, -1))) # [0, -1, -2, -3, -4, -5, -6, -7, -8, -9]
如果非法地傳入非整數(shù)的參數(shù),如:
print(list(range(10, 0.3)))
則會報以下的TypeError:
'float' object cannot be interpreted as an integer
最后提一下,我們常常會寫下如下代碼:
for i in range(10):
print(i)
此時Python解釋器實(shí)質(zhì)上會將range對象隱式轉(zhuǎn)化為迭代器,等價于如下代碼:
list_iterator = iter(range(10))
try:
while True:
x = next(list_iterator)
print(x)
except StopIteration:
pass
2. numpy.arange
numpy.arange是NumPy包的一個函數(shù),它的功能與Python內(nèi)置的range類似,它的原型可以近似表示為:
numpy.arange(stop, dtype=None, like=None) numpy.arange(start, stop, step=1, dtype=None, like=None)
(還是如前面所說,Python是不允許定義兩個類初始化函數(shù)的,其實(shí)其CPython實(shí)現(xiàn)更像是傳入不定長參數(shù)*args,然后根據(jù)len(args)來進(jìn)行不同的拆分,但我們這里遵循Python文檔風(fēng)格寫法)
其中start、step、step的使用與range類似,此處不再贅述,唯一的區(qū)別就是這3個參數(shù)都可以是小數(shù)。dtype為返回array的類型,如果沒有給定則會從輸入輸入?yún)?shù)中推斷。like為一個array-like的類型,它允許創(chuàng)建非NumPy arrays的arrays類型。
總結(jié)一下,該類與Python內(nèi)置的range區(qū)別有兩點(diǎn):一是支持小數(shù)參數(shù),二是返回ndarray類型而非像range那樣常常做為(隱式轉(zhuǎn)換為)list類型使用。
以下是其常見用例:
print(np.arange(3)) # [0 1 2] print(np.arange(3.0)) # [0. 1. 2.] print(np.arange(3,7)) # [3 4 5 6] print(np.arange(3,7,2)) # [3 5] print(np.arange(0, 5, 0.5)) #[0. 0.5 1. 1.5 2. 2.5 3. 3.5 4. 4.5]
注意,在numpy.arange的使用過程中可能存在浮點(diǎn)穩(wěn)定性的問題,從而導(dǎo)致下面這樣的意想不到的結(jié)果:
print(np.arange(0, 5, 0.5, dtype=int)) # [0 0 0 0 0 0 0 0 0 0] print(np.arange(-3, 3, 0.5, dtype=int)) # [-3 -2 -1 0 1 2 3 4 5 6 7 8]
這是因?yàn)樵?code>np.arange的內(nèi)部實(shí)現(xiàn)中,實(shí)際上的step值是按照公式dtype(start+step)-dtype(start)來計算的,而非直接采用step。當(dāng)進(jìn)行強(qiáng)制類型轉(zhuǎn)換(上面例子中轉(zhuǎn)為int,即朝0方向取整)或start遠(yuǎn)遠(yuǎn)比step大時,會出現(xiàn)精度的損失。在這種情況下,建議使用下面提到的np.linspace:
3. numpy.linspace
numpy.linspace也是Numpy內(nèi)置的一個函數(shù),它和numpy.arange類似,但是它不再是簡單的[start, stop)左閉右開,也沒有使用步長step,而是使用樣本個數(shù)num。其函數(shù)原型如下:
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
其中當(dāng)endpoint采用默認(rèn)的True時,start和stop表示序列的開始和初始值(閉區(qū)間[start, stop]),num為區(qū)間[start, stop]按照均勻(evenly)劃分采樣的樣本數(shù)(包括邊界start和stop在內(nèi))。不過需要注意的是,endpoint為True時stop才能做為最后一個樣本,為False時區(qū)間內(nèi)便不包括stop,此時會在區(qū)間[start,end]內(nèi)按照總個數(shù)為num + 1個樣本采樣并去掉尾部樣本(即stop點(diǎn))組成。retstep位置為True則會返回(samples, step)元組,其中samples為生成的樣本,step為樣本之間的間隔步長。
注意,它的start、stop參數(shù)都可以為小數(shù),但是當(dāng)dtype設(shè)置為int時則就不能為小數(shù)。
numpy.linspace的常見使用樣例如下:
print(np.linspace(2.0, 3.0, num=5)) # array([2. , 2.25, 2.5 , 2.75, 3. ])
如果設(shè)置endpoint為True,則按照num+1個樣本數(shù)量來采樣,并去掉最后一個樣本。
print(np.linspace(2.0, 3.0, num=5, endpoint=False)) # [2. 2.2 2.4 2.6 2.8]
如果retstep設(shè)置為True,則除了返回生成的樣本,還會返回樣本之間的間隔步長。
print(np.linspace(2.0, 3.0, num=5, retstep=True)) # (array([2. , 2.25, 2.5 , 2.75, 3. ]), 0.25)
下面我們用圖形形象化地描述endpoint取True和取False的區(qū)別:
import matplotlib.pyplot as plt N = 8 y = np.zeros(N) x1 = np.linspace(0, 10, N, endpoint=True) x2 = np.linspace(0, 10, N, endpoint=False) plt.plot(x1, y, 'o', color='orange') plt.plot(x2, y + 0.5, 'o', color='blue') plt.ylim([1, -0.5]) plt.show()
圖像顯示如下:

可以看出橘色的點(diǎn)為np.linspace(0, 10, N, endpoint=True),按照總共8個點(diǎn)在[0, 10]采樣,并包括stop邊界10。藍(lán)色的點(diǎn)為np.linspace(0, 10, N, endpoint=False),先按照總共9個點(diǎn)在[0, 10]采樣最后再去掉最后一個點(diǎn)(即stop點(diǎn)10),最終得到間隙更密的8個點(diǎn)。
參考
- [1] https://docs.python.org/3/library/stdtypes.html?highlight=range#range
- [2] https://stackoverflow.com/questions/43999181/range-non-default-parameter-follows-default-one
- [3] https://numpy.org/doc/stable/reference/generated/numpy.arange.html?highlight=arange#numpy.arange
- [4] https://numpy.org/doc/stable/reference/generated/numpy.linspace.html#numpy.linspace
到此這篇關(guān)于Python中range、np.arange和np.linspace的區(qū)別的文章就介紹到這了,更多相關(guān)Python range np.arange np.linspace內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Pytorch模型遷移和遷移學(xué)習(xí),導(dǎo)入部分模型參數(shù)的操作
這篇文章主要介紹了Pytorch模型遷移和遷移學(xué)習(xí),導(dǎo)入部分模型參數(shù)的操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
Python查找字符串中包含的多個元素的實(shí)現(xiàn)
本文詳細(xì)介紹了如何使用Python查找字符串中包含的多個元素,包括基本字符串操作和使用正則表達(dá)式進(jìn)行高級搜索,具有一定的參考價值,感興趣的可以了解一下2024-03-03
django 鏈接多個數(shù)據(jù)庫 并使用原生sql實(shí)現(xiàn)
這篇文章主要介紹了django 鏈接多個數(shù)據(jù)庫 并使用原生sql實(shí)現(xiàn),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
使用 Python 實(shí)現(xiàn)簡單的 switch/case 語句的方法
這篇文章主要介紹了用 Python 實(shí)現(xiàn)簡單的 switch/case 語句的方法,本文通過實(shí)例代碼給大家介紹的非常詳細(xì),具有一定的參考借鑒價值,需要的朋友可以參考下2018-09-09
Python使用Marshmallow輕松實(shí)現(xiàn)序列化和反序列化
這篇文章主要為大家詳細(xì)介紹了Python如何使用Marshmallow輕松實(shí)現(xiàn)序列化和反序列化,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解下2025-03-03
Python函數(shù)式編程之面向過程面向?qū)ο蠹昂瘮?shù)式簡析
這一番我們要學(xué)習(xí)點(diǎn)有難度的了,因此將降低閱讀與理解難度,盡量采用大白話為你鋪墊,因?yàn)樯婕暗囊恍└拍钜彩墙梃b的其它編程語言的風(fēng)格,而且實(shí)際落地中存在部分爭議不過多學(xué)一點(diǎn),總是沒有壞處的2021-09-09
Pandas借助Numpy實(shí)現(xiàn)優(yōu)化的條件檢索代碼
Numpy其實(shí)是最早的處理數(shù)據(jù)的Python庫,它的核心ndarray對象,是一個高效的n維數(shù)組結(jié)構(gòu),本文主要介紹了Pandas如何借助Numpy優(yōu)化條件檢索,感興趣的可以了解下2024-03-03

