Python迭代器&生成器

菜鸟教程-Python3 迭代器与生成器

1. 迭代器

迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

迭代器有两个基本的方法：iter() 和 next()

  1. nums = [1, 2, 3, 4]
  3. it = iter(nums)
  5. print(it)
  6. print(next(it))
  7. print(next(it))
  8. print(next(it))
  9. print(next(it))
  10. print(next(it))
  12. # output:
  13. # <list_iterator object at 0x0000028510E76C10>
  14. # 1
  15. # 2
  16. # 3
  17. # 4
  18. # Traceback (most recent call last):
  19. #   File "test.py", line 10, in <module>      
  20. #     print(next(it))
  21. # StopIteration

通过for语句遍历迭代器：

nums = [1, 2, 3, 4]

it = iter(nums)
for x in it:
    print(x, end=' ')

# output:
# 1 2 3 4

2. 生成器

import sys

def fibonacci(n):
    a, b, counter = 0, 1, 0
    while True:
        if counter > n:
            return
        yield a
        a, b = b, a+b
        counter += 1

f = fibonacci(10)

while True:
    try:
        print(next(f), end=" ")
    except StopIteration:
        sys.exit()

跟普通函数不同的是，生成器是一个返回迭代器的函数，只能用于迭代操作，更简单点理解生成器就是一个迭代器。

在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。