实例：Python 中迭代器与生成器

作者：袖梨 2022-06-30

Python 中迭代器与生成器实例详解

投稿：lqh 字体：[增加 减小] 类型：转载 时间：2017-03-29 我要评论

这篇文章主要介绍了Python 中迭代器与生成器实例详解的相关资料,需要的朋友可以参考下

Python 中迭代器与生成器实例详解

本文通过针对不同应用场景及其解决方案的方式，总结了Python中迭代器与生成器的一些相关知识，具体如下：

1.手动遍历迭代器

应用场景：想遍历一个可迭代对象中的所有元素，但是不想用for循环

解决方案：使用next()函数，并捕获StopIteration异常

?

1

2

3

4

5

6

7

8

9

10

defmanual_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

11

classNode:

  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

14

a=[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

15

classCountdown:

  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

3

withopen('/etc/passwd') as f:

  forlineinf:

    print(line,end='')

示例2

?

1

2

3

4

fromitertoolsimportdropwhile

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

7

fromcollectionsimportIterable

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)

相关文章

精彩推荐