• 五、迭代器和生成器综合例子
    • 1、反向迭代
    • 2、同时迭代多个序列

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

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

    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() 是可以接受多于两个的序列的参数,不仅仅是两个。