学习python的第十一天
今天学的这个东西叫generator,要创建一个generator很简单,只要把一个列表生成式的[]
改成()
,就创建了一个generator:
>>> 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称为生成器,举个例子,假设我们要创建一个含10万元素的列表,但是仅仅需要访问前前几个元素,这太不划算了,白白浪费了很大的空间
所以,如果列表元素可以按照某种算法推算出来,那我们是否可以在循环的过程中不断推算出后续的元素呢?这样就不必创建完整的list,从而节省大量的空间。在Python中,这种一边循环一边计算的机制,就是generator!
那么,如何打印generator的每一个元素呢,用next()
j函数获得generator的下一个返回值。
>>> next(g)
0
>>> next(g)
1
>>> next(g)
4
>>> next(g)
9
>>> next(g)
16
>>> next(g)
25
>>> next(g)
36
>>> next(g)
49
>>> next(g)
64
>>> next(g)
81
>>> next(g)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
为什么会报错呢?因为算完了,没有元素了,generator保存的是算法,当没有更多的元素时,就会抛出StopIteration
的错误。
当然,一个个的调用next(g)
太变态了,像大家这样的聪明人都知道用for
循环来解决这以问题,因为generator也是可迭代对象:
>>> g = (x * x for x in range(10))
>>> for n in g:
... print(n)
...
0
1
4
9
16
25
36
49
64
81
并且通过for循环来迭代它,就不需要关心StopIteration
的错误。
另外一种定义generator的方法:如果一个函数的定义中包含yield
关键词,那么这个函数就不是一个普通的函数,而是一个generator
def aaa():
for x in aaa():
while i<0:
print(1)
---------------------------------
def aaa():
for x in aaa():
while i<0:
yield 1
没错,我们只把print()
换成了yield
,函数aaa
就变成了generator
至于yield
是什么意思,怎么用,大家可以看一下这一位大佬的文章,我怕我解释不清楚yield用法
来看一下这个——杨辉三角
把每一行看做一个list,试写一个generator,不断输出下一行的list:
大家一起试试看!