Python 中迭代器與生成器實(shí)例詳解
本文通過針對(duì)不同應(yīng)用場(chǎng)景及其解決方案的方式,總結(jié)了Python中迭代器與生成器的一些相關(guān)知識(shí),具體如下:
1.手動(dòng)遍歷迭代器
應(yīng)用場(chǎng)景:想遍歷一個(gè)可迭代對(duì)象中的所有元素,但是不想用for循環(huán)
解決方案:使用next()函數(shù),并捕獲StopIteration異常
def manual_iter():
with open('/etc/passwd') as f:
try:
while True:
line=next(f)
if line is None:
break
print(line,end='')
except StopIteration:
pass
#test case
items=[1,2,3]
it=iter(items)
next(it)
next(it)
next(it)
2.代理迭代
應(yīng)用場(chǎng)景:想直接在一個(gè)包含有列表、元組或其他可迭代對(duì)象的容器對(duì)象上執(zhí)行迭代操作
解決方案:定義一個(gè)iter()方法,將迭代操作代理到容器內(nèi)部的對(duì)象上
示例:
class Node:
def __init__(self,value):
self._value=value
self._children=[]
def __repr__(self):
return 'Node({!r})'.fromat(self._value)
def add_child(self,node):
self._children.append(node)
def __iter__(self):
#將迭代請(qǐng)求傳遞給內(nèi)部的_children屬性
return iter(self._children)
#test case
if __name='__main__':
root=Node(0)
child1=Node(1)
child2=Nide(2)
root.add_child(child1)
root.add_child(child2)
for ch in root:
print(ch)
3.反向迭代
應(yīng)用場(chǎng)景:想要反向迭代一個(gè)序列
解決方案:使用內(nèi)置的reversed()函數(shù)或者在自定義類上實(shí)現(xiàn)reversed()
示例1
a=[1,2,3,4]
for x in reversed(a):
print(x) #4 3 2 1
f=open('somefile')
for line in reversed(list(f)):
print(line,end='')
#test case
for rr in reversed(Countdown(30)):
print(rr)
for rr in Countdown(30):
print(rr)
示例2
class Countdown:
def __init__(self,start):
self.start=start
#常規(guī)迭代
def __iter__(self):
n=self.start
while n > 0:
yield n
n -= 1
#反向迭代
def __reversed__(self):
n=1
while n <= self.start:
yield n
n +=1
4.有選擇的迭代
應(yīng)用場(chǎng)景:想遍歷一個(gè)可迭代對(duì)象,但是對(duì)它開始的某些元素并不感興趣,想跳過
解決方案:使用itertools.dropwhile()
示例1
with open('/etc/passwd') as f:
for line in f:
print(line,end='')
示例2
from itertools import dropwhile
with open('/etc/passwd') as f:
for line in dropwhile(lambda line:line.startwith('#'),f):
print(line,end='')
5.同時(shí)迭代多個(gè)序列
應(yīng)用場(chǎng)景:想同時(shí)迭代多個(gè)序列每次分別從一個(gè)序列中取一個(gè)元素
解決方案:使用zip()函數(shù)
6.不同集合上元素的迭代
應(yīng)用場(chǎng)景:想在多個(gè)對(duì)象執(zhí)行相同的操作,但是這些對(duì)象在不同的容器中
解決方案:使用itertool.chain()函數(shù)
7.展開嵌套的序列
應(yīng)用場(chǎng)景:想將一個(gè)多層嵌套的序列展開成一個(gè)單層列表
解決方案:使用包含yield from語句的遞歸生成器
示例
from collections import Iterable
def flatten(items,ignore_types=(str,bytes)):
for x in items:
if isinstance(x,Iterable) and not isinstance(x,ignore_types):
yield from flatten(x)
else:
yield x
#test case
items=[1,2,[3,4,[5,6],7],8]
for x in flatten(items):
print(x)
感謝閱讀,希望能幫助到大家,謝謝大家對(duì)本站的支持!
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索:z360901061
微信掃一掃加我為好友
QQ號(hào)聯(lián)系: 360901061
您的支持是博主寫作最大的動(dòng)力,如果您喜歡我的文章,感覺我的文章對(duì)您有幫助,請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點(diǎn)擊下面給點(diǎn)支持吧,站長(zhǎng)非常感激您!手機(jī)微信長(zhǎng)按不能支付解決辦法:請(qǐng)將微信支付二維碼保存到相冊(cè),切換到微信,然后點(diǎn)擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元

