「每天3分钟学Python」Python 列表生成式的进化:生成器

01

列表生成式的缺点

通过列表生成式,我们可以直接创建一个列表。但是,内存数量有限,列表容量肯定不能超过内存大小。

再有,创建一个包含100万个元素的列表,不仅占用很大的存储空间,如果我们仅仅需要访问前面10%的元素,那后面绝大多数元素占用的空间都白白浪费了。

02

解决办法

如果列表元素中的元素可以按照某种算法推算出来,那是否可以在循环过程中,推算出我们需要的一定数量的元素呢?

这样地话,我们就可以灵活地创建需要数量的list,从而节省大量的空间。

在Python中,这种一边循环一边计算的机制,称为生成器:generator

03

创建generator

下面介绍两种创建generator的方法

第一种方法很简单,只要把一个列表生成式的 [] 改成 (),就创建了一个generator

g = ( x * x for x in range(10) )

g is a generator object

第二种方法是函数中带有yield,那么此函数就不再是函数了,而是一个generator,

def generatorfun():

print('step 1')

yield(10)

print('step 2')

yield 30

python的功能的确很强大,这使得它成为人工智能,AI ,机器学习,深度学习的主要用语言。