版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kabuto_hui/article/details/87898429
学习于 廖雪峰的官方网站
1. 生成器
在Python中,这种一边循环一边计算的机制,称为生成器:generator。
创建生成器的方法有两种:
-
将列表生成式的
[]
改为()
-
使用关键字
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
时,发现拿不到generator
的return
语句的返回值。如果想要拿到返回值,必须捕获StopIteration
错误,返回值包含在StopIteration
的value
中:
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
对象。它们表示一个惰性计算的序列;
集合数据类型如list
、dict
、str
等是Iterable
但不是Iterator
,不过可以通过iter()
函数获得一个Iterator
对象。