迭代器

• 迭代器可以记住遍历位置的对象
• list、tuple、dict都只是可以迭代的对象，不是迭代器，所以要创建一个迭代器
• 迭代器的俩个基本方法 iter() 和 next()
• iter() 用于创建一个迭代器、next() 输出当前位置的对象，只能往前不能后退
• 从一个简单的例子开始，list 输出前俩个元素

#!/usr/bin/python3
num = [1, 2, 3, 4, 5, 6, 7, 8, 9]
it = iter(num)  #创建迭代器
print(next(it))
print(next(it))
#输出结果
1
2

• 如何用于for循环呢

#!/usr/bin/python3
num = [1, 2, 3, 4, 5, 6, 7, 8, 9]
it = iter(num)
for x in it:
    print(x)
#输出结果
1
2
3
4
5
6
7
8
9

• 使用next()方法来遍历呢
• next()遍历到最后的时候会抛出StopIteration异常，捕获然后跳出循环

#!/usr/bin/python3
num = [1, 2, 3, 4, 5, 6, 7, 8, 9]
it = iter(num)
while True:
    try:
        print(next(it))
    except StopIteration:
        break
#输出结果 同上

生成器

• 生成器是迭代器的一种
• 使用yield函数 就像单步调试一样记住当前执行的位置，使用next()之后会从原来的位置继续执行。
• 先来个简单的例子看下怎么使用

#!/usr/bin/python3
def foo():
    while True:
        yield 4
        print('end....')

f = foo()
print(next(f))
print('中间')
print(next(f))
#输出结果
4
中间
end....
4

• 使用def 定一个了一个函数foo，单独看这个函数while循环一直是true，这里先把yield 看成return返回4
• f = foo() 将函数赋值给了f
• 执行print(next(f)) 会先输出4
• 由于使用了yield 所以函数不会继续执行
• 那就执行print('中间') 输出了 ‘中间’
• print(next(f)) 这次再次执行会在原来的执行位置继续执行 print('end....')
• 最后while继续执行输出 4

总结一下：其实从例子完全可以看出， yield会返回一个结果，并且记住当前执行的位置，通过使用next() 会在原来执行的位置继续执行程序。

来一个稍微复杂点的，使用生成器实现斐波那契数列

#!/usr/bin/python3
def foo(n):
    a, b, item = 0, 1, 1
    while True:
        if(item > n):
            return
        yield b
        a, b = b, a+b
        item+=1

f = foo(10)
while True:
    try:
        print(next(f))
    except StopIteration:
        print('end...')
        break

#输出结果
1
1
2
3
5
8
13
21
34
55
end...