Python迭代器和生成器

迭代器

迭代器是访问集合元素的一种方法；是可以记住遍历的位置的对象。

迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问；他有两个基本的方法，iter()和next()

字符串，列表或远足对象都可以用于创建迭代器

>>> list1=[1,2,3,4]
>>> it1=iter(list1) #创建迭代器对象
>>> print(next(it1))
1
>>> print(next(it1))
2
>>> print(next(it1))
3
>>> print(next(it1))
4
>>> print(next(it1))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

可以使用常规的for进行迭代器对象遍历

>>> list1=[1,2,3,4]                
>>> it1=iter(list1) #创建迭代器对象
>>> for x in it1:
...     print(x,end=",")
... 
1,2,3,4,>>> 

也可以

#!/usr/bin/env python

import sys

list1 = [1,2,3,4]

it1 = iter(list1)

while True:
    try:
        print(next(it1))
    except StopIteration:
        sys.exit()

结果

1
2
3
4

生成器

使用了 yield 的函数被称为生成器（generator）；跟普通的函数不同，生成器是一个返回迭代器的函数，只能用于迭代操作；在调用生成器运行的过程中，每次遇到yield时候会暂停并保存当前所有的运行信息，返回yield的值。并在下次执行next()方法时从当前位置继续

#!/usr/bin/env python

import sys

def fibonacci(n): #生成器函数
    a,b,cnt = 0,1,0
    while True:
        if (cnt > n):
            return
        yield a
        a,b = b,a+b
        cnt +=1
f = fibonacci(10) # f是一个迭代器，由生成器返回生成

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

结果

0,1,1,2,3,5,8,13,21,34,55,