Python学习笔记之生成器与迭代器

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kabuto_hui/article/details/87898429


学习于 廖雪峰的官方网站

1. 生成器

在Python中,这种一边循环一边计算的机制,称为生成器:generator。

创建生成器的方法有两种:

  1. 将列表生成式的[]改为()

  2. 使用关键字yield

1. 创建生成器方法1-[]->()

g = (x*x for x in range(5))
print(g)
print(next(g))
print(next(g))
print(next(g))
<generator object <genexpr> at 0x0000017F69EA6F48>
0
1
4

generator保存的是算法,每次调用next(g),就计算出g的下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration的错误。所以一般用循环方式来访问生成器:

g = (x*x for x in range(5))
for i in g:
    print(i)
0
1
4
9
16

1.2 创建生成器的方法2-关键字yield

一下面的斐波那契函数为例子:

def fib(max):
    n, a, b = 0, 0, 1
    while n < max:
        yield b
        a, b = b, a + b
        n = n + 1
    return 'done'

在上面的函数中,有一个yield关键字,这个关键字将这个函数变成了一个生成器。generator和函数的执行流程不一样。函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。

a = fib(5)
for i in a:
    print(i)
1
1
2
3
5

但是用for循环调用generator时,发现拿不到generatorreturn语句的返回值。如果想要拿到返回值,必须捕获StopIteration错误,返回值包含在StopIterationvalue中:

a = fib(5)
while True:
    try:
        print(next(a))
    except StopIteration as e:
              print(e.value)
              break
1
1
2
3
5
done

1.3 生成器例子-杨辉三角

杨辉三角定义如下:

扫描二维码关注公众号,回复: 5754900 查看本文章
      1
     / \
    1   1
   / \ / \
  1   2   1
 / \ / \ / \
1   3   3   1
def triangles():
    L = [1]
    while True:
        yield L
        L = [1] + [L[i] + L[i+1] for i in range(len(L)-1)] +[1]

count = 0
for i in triangles():
    print(i)
    count+=1
    if count == 10:
        break
        
[1]
[1, 1]
[1, 2, 1]
[1, 3, 3, 1]
[1, 4, 6, 4, 1]
[1, 5, 10, 10, 5, 1]
[1, 6, 15, 20, 15, 6, 1]
[1, 7, 21, 35, 35, 21, 7, 1]
[1, 8, 28, 56, 70, 56, 28, 8, 1]
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]

2. 迭代器

凡是可作用于for循环的对象都是Iterable类型,即可迭代类型;
凡是可作用于next()函数的对象都是Iterator类型,即为迭代器,所以生成器都是Iterator对象。它们表示一个惰性计算的序列;
集合数据类型如listdictstr等是Iterable但不是Iterator,不过可以通过iter()函数获得一个Iterator对象。

猜你喜欢

转载自blog.csdn.net/kabuto_hui/article/details/87898429