五、迭代器和生成器综合例子

因为迭代器和生成器基本是互通的,因此有些知识点需要综合在一起

1、反向迭代

反向迭代,应该也是常有的需求了,比如从一开始迭代的例子里,有个输出 list 的元素,从 1 到 5 的

  1. list1 = [1,2,3,4,5]
  2. for num1 in list1 :
  3. print ( num1 , end = ' ' )

那么我们从 5 到 1 呢?这也很简单, Python 中有内置的函数 reversed()

  1. list1 = [1,2,3,4,5]
  2. for num1 in reversed(list1) :
  3. print ( num1 , end = ' ' )

方向迭代很简单,可是要注意一点就是:反向迭代仅仅当对象的大小可预先确定或者对象实现了 __reversed__() 的特殊方法时才能生效。 如果两者都不符合,那你必须先将对象转换为一个列表才行

其实很多时候我们可以通过在自定义类上实现 __reversed__() 方法来实现反向迭代。不过有些知识点在之前的篇节中还没有提到,不过可以相应的看下,有编程基础的,学完上面的知识点应该也能理解的。

  1. # -*- coding: UTF-8 -*-
  2. class Countdown:
  3. def __init__(self, start):
  4. self.start = start
  5. def __iter__(self):
  6. # Forward iterator
  7. n = self.start
  8. while n > 0:
  9. yield n
  10. n -= 1
  11. def __reversed__(self):
  12. # Reverse iterator
  13. n = 1
  14. while n <= self.start:
  15. yield n
  16. n += 1
  17. for rr in reversed(Countdown(30)):
  18. print(rr)
  19. for rr in Countdown(30):
  20. print(rr)

输出的结果是 1 到 30 然后 30 到 1 ,分别是顺序打印和倒序打印

2、同时迭代多个序列

你想同时迭代多个序列,每次分别从一个序列中取一个元素。你遇到过这样的需求吗?

为了同时迭代多个序列,使用 zip() 函数,具体示例:

  1. # -*- coding: UTF-8 -*-
  2. names = ['laingdianshui', 'twowater', '两点水']
  3. ages = [18, 19, 20]
  4. for name, age in zip(names, ages):
  5. print(name,age)

输出的结果:

  1. laingdianshui 18
  2. twowater 19
  3. 两点水 20

其实 zip(a, b) 会生成一个可返回元组 (x, y) 的迭代器,其中 x 来自 a,y 来自 b。 一旦其中某个序列到底结尾,迭代宣告结束。 因此迭代长度跟参数中最短序列长度一致。注意理解这句话喔,也就是说如果 a , b 的长度不一致的话,以最短的为标准,遍历完后就结束。

利用 zip() 函数,我们还可把一个 key 列表和一个 value 列表生成一个 dict (字典),如下:

  1. # -*- coding: UTF-8 -*-
  2. names = ['laingdianshui', 'twowater', '两点水']
  3. ages = [18, 19, 20]
  4. dict1= dict(zip(names,ages))
  5. print(dict1)

输出如下结果:

  1. {'laingdianshui': 18, 'twowater': 19, '两点水': 20}

这里提一下, zip() 是可以接受多于两个的序列的参数,不仅仅是两个。