手记

Python列表推导式与生成器

愿你一直充满无限的潜力


列表推导式

>>> [x+1 for x in range(10) ]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

从上面的代码可以看出,列表推导式是生成一个列表。

生成器

>>> g=(x+1 for x in range(10))>>> g
<generator object <genexpr> at 0x7f3af91b7570>>>> next(g)1>>> next(g)2>>> next(g)3>>> next(g)4>>> next(g)5>>> next(g)6>>> next(g)7>>> next(g)8>>> next(g)9>>> next(g)10>>> next(g)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>StopIteration>>>

从上面的代码可以看出,生成器的形式和列表推导式差不多,只是括号的不同。

本质区别是列表推导式是一次性生成,而生成器是通过相应的算法推导出来的值,在数据量较大的时候,你不可能一次性生成那么大的列表,生成器是最好的选择。

例如:用生成器产生斐波拉契数列

def fibs(max):
    n,a,b=0,0,1
    while n < max:        yield b
        a,b = b,a+b
        n+=1for n in fibs(6):
    print(n)



作者:olaH
链接:https://www.jianshu.com/p/ae31823c9acd


0人推荐
随时随地看视频
慕课网APP