普通的迭代对象所有数据都在内存中,如果数据海量的话将会非常耗内存;生成器函数被调用后,其函数体内的代码并不会立即执行,而是返回一个生成器。当返回的生成器调用成员方法时,相应的生成器函数中的代码才会执行
迭代
如果可以从一个对象中,逐个地获取元素,那么就说这个对象是可迭代的
#列表是可迭代的,类似的还有字符串、文件等
for i in [1, 2, 3, 4]:
print(i)
迭代器是一种只允许迭代一次的对象,通过调用 next()
方法依次获取下一个元素;当迭代器中没有元素时,调用 next()
方法会抛出
StopIteration
异常,捕获到异常时退出
生成器
生成器是特殊的迭代器,包含 yield
表达式的函数是一个生成器函数
Python在处理 for
循环时,首先会调用内建函数 iter(something)
,实际上调用 something.__iter__()
,返回
something
对应的迭代器。而后,for
循环会调用内建函数 next()
,作用在迭代器上,获取迭代器的下一个元素,并赋值给 x
。
此后,Python 才开始执行循环体,yield相当于return,下次迭代时,代码从yield的下一跳语句开始执行
def yield_test(n):
for i in range(n):
yield i**2
print("i=",i)
print("end...")
for i in yield_test(5):
print(i,",")
0 ,
i= 0
1 ,
i= 1
4 ,
i= 2
9 ,
i= 3
16 ,
i= 4
end...