Python中的迭代
在Python中迭代序列(或者其他可迭代對象)時(shí),有一些函數(shù)非常好用。有些函數(shù)位于itertools模塊中,還有一些Python的內(nèi)建函數(shù)也十分方便。
1. 并行迭代
程序可以同時(shí)迭代兩個(gè)序列。比如有下面兩個(gè)列表:
names = ['anne', 'beth', 'george', 'damon'] ages = [12, 45, 32, 102]如果想要打印名字和對應(yīng)的年齡,可以像下面這樣做:
''' 遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:579817333 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' In [7]: for i in range(len(names)):...: print(names[i], 'is', ages[i], 'years old')...: anne is 12 years old beth is 45 years old george is 32 years old damon is 102 years old這里 i 是循環(huán)索引的標(biāo)準(zhǔn)變量名。
而內(nèi)建的zip函數(shù)就可以用來進(jìn)行并行迭代,可以把兩個(gè)序列 “壓縮” 在一起,然后返回一個(gè)元組的列表:
>>> zip(names, ages) [('anne', 12), ('beth', 45), ('george', 32), ('damon', 102)]現(xiàn)在我可以在循環(huán)中解包元組:
''' 遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:579817333 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' In [9]: for name, age in zip(names, ages):...: print(name, 'is', age, 'years old')...: anne is 12 years old beth is 45 years old george is 32 years old damon is 102 years oldzip 函數(shù)也可以作用于任意多的序列。關(guān)于它很重要的一點(diǎn)是zip可以處理不等長的序列,當(dāng)最短的序列 “用完” 的時(shí)候就會停止:
>>> zip(range(5), xrange(1000000000)) [(0, 0), (1, 1), (2, 2), (3, 3), (4, 4)]在上面的代碼中,不推薦用range替換xrange——盡管只需要要前5個(gè)數(shù)字,但range會計(jì)算所有的數(shù)字,這里要花費(fèi)很長的時(shí)間。而使用xrange就沒有這個(gè)問題,它只計(jì)算前5個(gè)數(shù)字。
2. 按索引迭代
有些時(shí)間想要迭代訪問序列中的對象,同時(shí)還有獲取當(dāng)前對象的索引。例如,在一個(gè)字符串列表中替換所有包含’xxx’的子字符串。實(shí)現(xiàn)的方法肯定有很多,假設(shè)你想象下面這樣做:
for string in strings:if 'xxx' in string:index = strings.index(string) # Search for the string in the list of stringsstrings[index] = '[censored]'沒問題,但是在替換前要搜索給定的字符串似乎沒必要。如果不替換的話,搜索還會返回錯(cuò)誤的索引(前面出現(xiàn)的同一個(gè)詞的索引)。一個(gè)比較好的版本如下:
''' 遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:579817333 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' index = 0 for string in strings:if 'xxx' in string:strings[index] = '[censored]'index += 1方法有些笨,不過還可以接受。另一種方法是使用內(nèi)建的enumerate函數(shù):
for index, string in enumerate(strings):if 'xxx' in string:strings[index] = '[censored]'這個(gè)函數(shù)可以在提供索引的地方迭代索引-值對
3. 翻轉(zhuǎn)和排序迭代
讓我們看看另外兩個(gè)有用的函數(shù):reversed和sorted。它們同列表的reverse和sort(sorted和sort使用同樣的參數(shù))方法類似,但作用于任何序列或可迭代對象上,不是原地修改對象,而是返回翻轉(zhuǎn)或排序后的版本:
>>> sorted([4, 3, 6, 8, 3]) [3, 3, 4, 6, 8] >>> sorted('Hello, world!') [' ', '!', ',', 'H', 'd', 'e', 'l', 'l', 'l', 'o', 'o', 'r', 'w'] >>> list(reversed('Hello, world!')) ['!', 'd', 'l', 'r', 'o', 'w', ' ', ',', 'o', 'l', 'l', 'e', 'H'] >>> ''.join(reversed('Hello, world!')) '!dlrow ,olleH'注意,雖然sorted方法返回列表,reversed方法卻返回一個(gè)更加不可思議的可迭代對象。它們具體的含義不用過多關(guān)注,大可在for循環(huán)以及join方法中使用,而不會有任何問題。不過卻不能直接對它使用索引、分片以及調(diào)用list方法,如果希望進(jìn)行上述處理,那么可以使用list類型轉(zhuǎn)換返回的對象,上面的例子中已經(jīng)給出具體的做法。
4. 迭代器規(guī)則
迭代的意思是重復(fù)做一些事很多次,就像在循環(huán)中做的那樣。到現(xiàn)在為止只在for循環(huán)中對序列和字典進(jìn)行過迭代,但實(shí)際上也能對其他對象進(jìn)行迭代:只要改對象實(shí)現(xiàn)了__ iter __ 方法。
__ iter __ 方法會返回一個(gè)迭代器(iterator),所謂的迭代器就是具有next方法(這個(gè)方法在調(diào)用時(shí)不需要任何參數(shù))的對象。在調(diào)用next方法時(shí),迭代器會返回它的下一個(gè)值。如果next方法被調(diào)用,但迭代器沒有值可以返回,就會引發(fā)一個(gè)StopIteration異常。
注意:迭代器規(guī)則在Python 3.0中有一些變化。在新的規(guī)則中,迭代器對象應(yīng)該實(shí)現(xiàn) __ next __
方法,而不是next。而新的內(nèi)建函數(shù)next可以用于訪問這個(gè)方法。換句話說,next(it)等同于3.0之前版本中的it.next()。
迭代規(guī)則的關(guān)鍵是什么?為什么不使用列表?因?yàn)榱斜淼臍μ蟆H绻粋€(gè)函數(shù),可以一個(gè)接一個(gè)地計(jì)算值,那么在使用時(shí)可能是計(jì)算一個(gè)值時(shí)獲取一個(gè)值——而不是通過列表一次性獲取所有值。如果有很多值,列表就會占用太多的內(nèi)存。但還有其他的理由:使用迭代器更通用、更簡單、更優(yōu)雅。讓我們看看一個(gè)不使用列表的例子,因?yàn)橐玫脑?#xff0c;列表的長度必須無限。
這里的“列表”是一個(gè)斐波那契數(shù)列。使用的迭代器如下:
''' 遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:579817333 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' In [1]: class Fibs:...: def __init__(self):...: self.a = 0...: self.b = 1...: def next(self):...: self.a, self.b = self.b, self.a + self.b...: return self.a...: def __iter__(self):...: return self注意,迭代器實(shí)現(xiàn)了__ iter__ 方法,這個(gè)方法實(shí)際上返回迭代器本身。在很多情況下,__ iter__ 會放到其他的會在for循環(huán)中使用的對象中。這樣一來,程序就能返回所需的迭代器。此外,推薦使用迭代器實(shí)現(xiàn)它自己的 __ iter__ 方法,然后就能直接在for循環(huán)中使用迭代器本身了。
注意:正式的說法是,一個(gè)實(shí)現(xiàn)了__iter__方法的對象是可迭代的,一個(gè)實(shí)現(xiàn)了next方法的對象則是迭代器。
首先,產(chǎn)生一個(gè)Fibs對象:
In [2]: fibs = Fibs()可在for循環(huán)中使用該對象——比如去查找在斐波那契數(shù)列中比1000大的數(shù)中的最小的數(shù):
In [3]: for f in fibs:...: if f > 1000:...: print f...: break因?yàn)樵O(shè)置了break,所以循環(huán)在這里停止了,否則循環(huán)會一直繼續(xù)下去。
提示 內(nèi)建函數(shù)iter可以從可迭代的對象中獲得迭代器。
>>> it = iter([1, 2, 3]) >>> it.next() 1 >>> it.next() 2除此之外,它也可以從函數(shù)或者其他可調(diào)用對象中獲取可迭代對象。
5. 從迭代器得到序列
除了在迭代器和可迭代對象上進(jìn)行迭代外,還能把它們轉(zhuǎn)換為序列。在大部分能使用序列的情況下(除了或者分片等操作中),都能使用迭代器(或者可迭代對象)替換。關(guān)于這個(gè)的一個(gè)很有用的例子是使用list構(gòu)造方法顯示地將迭代器轉(zhuǎn)化為列表。
''' 遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:579817333 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' >>> class TestIterator: ... value = 0 ... def next(self): ... self.value += 1 ... if self.value > 10: ... raise StopIteration ... return self.value ... def __iter__(self): ... return self ... >>> ti = TestIterator() >>> list(ti) [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]總結(jié)
以上是生活随笔為你收集整理的Python中的迭代的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 内网穿透(端口映射)实践,让你的小伙伴访
- 下一篇: python项目实战分析:爬取时光网电影