最近在学习python的过程中接触了一个新的概念——生成器,我们都知道迭代器,c++,java中都有iterator的概念,那么生成器是什么概念呢,yield又究竟如何理解,下面让我来慢慢讲解。
首先,想要理解yield,我们应该先了解,和生成器的概念。
一、迭代
python中的迭代(iterator)的概念和c++中的类似,是一种取代下标访问的方式。通常我们用for来遍历的类都是使用迭代器实现的。例如:
list = [1,2,3]for i in list: #似乎和c++的 for(auto iter:vec){}很像 dosomething
二、生成器
有时我们需要对一个列表或字典进行计算并使用,可能我们会这么做
list = [1,2,3]g = [x*x for x in list]
这样会得到一个计算后的列表。如果这个列表非常大,那无疑会浪费很多空间。
生成器便为此而生,遍历的过程中计算,
gen = (x*x for x in list) #得到一个生成器 for i in gen
三、yield
函数中只要出现了yield语句就会将其转为一个生成器,其底层如何运转,让我们看一个python cookbook上的例子。
>>>def countdown(n):... print('Starting to count form ', n)... while n>0:... yield n... n-=1... print('Done!')>>>c = countdown(3)>>>c>>>next(c)Starting to count form 33>>>next(c)2>>>next(c)1>>>next(c)Done!Traceback...
获得一个生成器后,第一次调用时执行print语句,然后输出n,知道最后,执行 print('Done!')
通过yield 我们可以更加灵活的获得我们想要的一个生成器.