Python 中迭代器与生成器实例详解
投稿:lqh 字体:[增加 减小] 类型:转载 时间:2017-03-29 我要评论
这篇文章主要介绍了Python 中迭代器与生成器实例详解的相关资料,需要的朋友可以参考下
Python 中迭代器与生成器实例详解
本文通过针对不同应用场景及其解决方案的方式,总结了Python中迭代器与生成器的一些相关知识,具体如下:
1.手动遍历迭代器
应用场景:想遍历一个可迭代对象中的所有元素,但是不想用for循环
解决方案:使用next()函数,并捕获StopIteration异常
?1
2
3
4
5
6
7
8
9
10defmanual_iter():
withopen('/etc/passwd') as f:
try:
whileTrue:
line=next(f)
iflineisNone:
break
print(line,end='')
exceptStopIteration:
pass
?1
2
3
4
5
6#test case
items=[1,2,3]
it=iter(items)
next(it)
next(it)
next(it)
2.代理迭代
应用场景:想直接在一个包含有列表、元组或其他可迭代对象的容器对象上执行迭代操作
解决方案:定义一个iter()方法,将迭代操作代理到容器内部的对象上
示例:
?1
2
3
4
5
6
7
8
9
10
11classNode:
def__init__(self,value):
self._value=value
self._children=[]
def__repr__(self):
return'Node({!r})'.fromat(self._value)
defadd_child(self,node):
self._children.append(node)
def__iter__(self):
#将迭代请求传递给内部的_children属性
returniter(self._children)
?1
2
3
4
5
6
7
8
9#test case
if__name='__main__':
root=Node(0)
child1=Node(1)
child2=Nide(2)
root.add_child(child1)
root.add_child(child2)
forchinroot:
print(ch)
3.反向迭代
应用场景:想要反向迭代一个序列
解决方案:使用内置的reversed()函数或者在自定义类上实现reversed()
示例1
?1
2
3
4
5
6
7
8
9
10
11
12
13
14a=[1,2,3,4]
forxinreversed(a):
print(x)#4 3 2 1
f=open('somefile')
forlineinreversed(list(f)):
print(line,end='')
#test case
forrrinreversed(Countdown(30)):
print(rr)
forrrinCountdown(30):
print(rr)
示例2
?1
2
3
4
5
6
7
8
9
10
11
12
13
14
15classCountdown:
def__init__(self,start):
self.start=start
#常规迭代
def__iter__(self):
n=self.start
whilen >0:
yieldn
n-=1
#反向迭代
def__reversed__(self):
n=1
whilen <=self.start:
yieldn
n+=1
4.有选择的迭代
应用场景:想遍历一个可迭代对象,但是对它开始的某些元素并不感兴趣,想跳过
解决方案:使用itertools.dropwhile()
示例1
?1
2
3withopen('/etc/passwd') as f:
forlineinf:
print(line,end='')
示例2
?1
2
3
4fromitertoolsimportdropwhile
withopen('/etc/passwd') as f:
forlineindropwhile(lambdaline:line.startwith('#'),f):
print(line,end='')
5.同时迭代多个序列
应用场景:想同时迭代多个序列每次分别从一个序列中取一个元素
解决方案:使用zip()函数
6.不同集合上元素的迭代
应用场景:想在多个对象执行相同的操作,但是这些对象在不同的容器中
解决方案:使用itertool.chain()函数
7.展开嵌套的序列
应用场景:想将一个多层嵌套的序列展开成一个单层列表
解决方案:使用包含yield from语句的递归生成器
示例
?1
2
3
4
5
6
7fromcollectionsimportIterable
defflatten(items,ignore_types=(str,bytes)):
forxinitems:
ifisinstance(x,Iterable)andnotisinstance(x,ignore_types):
yieldfromflatten(x)
else:
yieldx
?1
2
3
4#test case
items=[1,2,[3,4,[5,6],7],8]
forxinflatten(items):
print(x)