>>> L = [x * x for x in range(10)]
>>> L
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x1022ef630>
生成器 generator
要创建一个generator,有很多种方法。第一种方法很简单,只要把一个列表生成式的[]
改成()
,就创建了一个generator,如上所示。如果要一个一个打印出来,可以通过next()
函数获得generator的下一个返回值。
我们讲过,generator保存的是算法,每次调用next(g)
,就计算出g
的下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration
的错误。
如果一个函数定义中包含yield
关键字,那么这个函数就不再是一个普通函数,而是一个generator。和return不一样的地方是,有yield的函数,在使用next()调用该函数时遇到yield返回,再一次调用时,从上次yield结束下一行继续执行,知道下一次遇到yield或return。
同样的,把函数改成generator后,我们基本上从来不会用next()
来获取下一个返回值,而是直接使用for
循环来迭代。
map(f, l) 传入的第一个参数是函数对象本身,l是一个序列。整体是一个Iterator,它是一个惰性序列。map()
作为高阶函数,事实上它把运算规则抽象了。
reduce(f, l) 接收两个参数,reduce
把结果继续和序列的下一个元素做累积计算。
from functools import reduce
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)