任何包含yield 语句的函数称为生成器。
yield不像return 返回值,而是每次产生多个值。每次产生一个值时,函数就会被冻结。被激活后从停止点开始.
def flatten(nested):
for sublist in nested:
for element in sublist:
yield element
>>> nested = [[1,2],[3,4],[5]]
>>> for num in flatten(nested):
print(num)
1
2
3
4
5
>>> list(flatten(nested))
[1, 2, 3, 4, 5]
数量少用列表推导式,数量多用 生成器。
生成器可以在当前的圆括号内直接使用,不用增加另外一对括号:
>>> sum(i**2 for i in range(10))
285