生成器表达式和推导式

生成器表达式和列表推导式非常相似,区别就是将方括号换成了小括号。

生成器表达式返回的是一个生成器迭代器 generator。表达式会产生一系列值,可使用内置函数 next() 逐一获取,或使用 for 循环遍历。

生成器迭代器中的元素,仅供一次性使用,而且可以要多少取多少,它会记住你取了多少,取到了哪里。

(i*i for i in range(3))
<generator object <genexpr> at 0x00000271080FBC00>
g = (i*i for i in range(5))
type(g)
generator
next(g), next(g)
(0, 1)
for i in g:
    print(i)
4
9
16

这种表达式被设计用于生成器将立即被外层函数所使用的情况。生成器表达式相比完整的生成器更紧凑但较不灵活,相比等效的列表推导式则更为节省内存。

sum(i*i for i in range(5))
30
list(i*i for i in range(5))
[0, 1, 4, 9, 16]

除了列表推导式和生成器表达式,集合和字典也可使用类似的推导式。

{i*2 for i in 'abac'}
{'aa', 'bb', 'cc'}
{i:j for i,j in zip('abc',(1,2,3))}
{'a': 1, 'b': 2, 'c': 3}
jupyter附件