Python:yield关键字

  最近在学习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
<generator object countdown at 0x1006a0af0>

>>>next(c)
Starting to count form 3
3

>>>next(c)
2

>>>next(c)
1

>>>next(c)
Done!

Traceback...

获得一个生成器后,第一次调用时执行print语句,然后输出n,知道最后,执行 print('Done!')

 

通过yield 我们可以更加灵活的获得我们想要的一个生成器.

  

 

posted @ 2017-03-15 18:26  爱吃土豆的男孩  阅读(228)  评论(0编辑  收藏  举报