python中的生成器(一)

咱们先考虑一个场景:python

  有个情景须要循环输出1——10.算法

  这里给两种方法:缓存

list1 = [1,2,3,4,5,6,7,8,9,10]
for i in list1:
    print(i)
for i in range(1,11):
    print(i)

两种方式输出结果同样,可是咱们考虑一下,若是要求输出1——1000000呢?函数

第一种方式会致使list1里面真实放入1000000长度的数字,占用空间很大,明显不是明智之举,spa

再来看第二种方法,用到range帮助咱们生成数据,在python3中range的本质就是一个生成器。.net

在python2中:range返回的是一个等差列表,好比[0,1,2,3,4,5,6,7,```````], 而xrange才是返回一个生成器对象. 即python2 range()==[```````````````````], python2 xrange()==python3 range()code

具体对比查看:https://blog.csdn.net/humanking7/article/details/45950967对象

(一)这里写一个函数,在生成器函数的名称中加上gen 前缀或后缀,不过这不是必要的习惯:blog

1 def gen_create_range(start,end):
2 
3     while start < end:
4         yield start
5         start += 1
6 
7 for i in gen_create_range(1,5):
8     print(i)

 #output:generator

       1

    2

    3

    4

  这个函数没有return 可是能够有返回值,注意看里面有个yield关键字,这个函数和range()函数很像。

  (二)什么是生成器:函数定义中包含yield关键字那么函数就变成了生成器

  概念:若是列表元素能够按照某种算法推算出来,那咱们是否能够在循环的过程当中不断推算出后续的元素呢?这样就没必要建立完整的list,从而节省大量的空间。在Python中,这种一边循环一边计算的机制,称为生成器:generator   

  生成器表达式: 通列表解析语法,只不过把列表解析的[]换成()  

g= (x**2 for x in range(5))

print g

>> 
<generator object <genexpr> at 0x0000000002771798>

 

#若是、
L=[x**2 for x in range(5)]

print L 

>>
[0, 1, 4, 9, 16]

  也就是说:建立L和g的区别仅在于最外层的[]和(),L是一个list,而g是一个generator

1).函数中只要出现了yield语句就会将其转变成一个生成器函数

  • 特别之处在于,yield 的做用就是把一个函数变成一个 generator,带有 yield 的函数再也不是一个普通函数,Python 解释器会将其视为一个 generator

  • 与普通函数不同,生成器值会在迭代操做的时候才能运行.yiled能够把函数中断,保存状态和继续执行的能力

比如一个武打片里面的慢镜头回放,yield把函数里面你要保存的值中断并保存,你经过调用next()来回放

 

好比:

def countdown(n):
   print('Starting to count from',n)
   while n>0:
      yield n
      n-=1
   print('done')

c=countdown(3)
print(c)
>>
<generator object countdown at 0x0000000002821828>
#表示这是一个生成器

 

 2).调用该generator时,首先要生成一个generator对象,而后用next()函数不断得到下一个返回值

好比:

c=countdown(3)

#run the first yield and emit a value

print(next(c))

>>

Starting to count from 3

3

#run the next yield

print(next(c))

>>

2

#run the next yield

print(next(c))

>>

1

#run the next yield

print(next(c))

>>

done

    print next(c)

StopIteration

 

深刻解释:

你把yield想象成时间断点,运行一次next就回放一下,看起来就好像一个函数在正常执行的过程当中被 yield 中断了数次,每次中断都会经过 yield 返回当前的迭代值

 

  • 第一次next()是打印了 print('Starting to count from',n),提取第一次的保存值是3

     

  • 第二次再运行next()是继续在while里面的断点接着走,因此没有打印print('Starting to count from',n), 而是直接提取第二次的保存值2

     

  • 第三次再运行next()是继续在while里面的断点接着走,因此直接输出1

     

  • 第四次再运行next()的时候,发现yield缓存的武打片慢镜头都已经放完了,因此输出done以后,报了个错StopIteration

 

3).生成器函数用for循环

for n in countdown(6):
  
  print(n)

>>

Starting to count from 6

6

5

4

3

2

1

done

 

 

正确的方法是使用for循环,由于generator也是可迭代对象

4yield 与 return相爱相杀

1).在一个生成器中,若是没有return,则默认执行到函数完毕时返回StopIteration

def gen1():
    yield 100

g1=gen1()
print(next(g1))
>>100

第一次调用next(g1)时,会在执行完yield语句后挂起,因此此时程序并无执行结束。

print(next(g1))
>>
Traceback (most recent call last):
  File "C:/about_gen.py", line 71, in <module>
    print(next(g1))
StopIteration

程序试图从yield语句的下一条语句开始执行,发现已经到告终尾,因此抛出StopIteration异常

 

2).若是遇到return,若是在执行过程当中 return,则直接抛出 StopIteration 终止迭代

def gen2():
    yield 200
    return 
    yield 300

g2=gen2()
print(next(g2))
>>200
# 程序停留在执行完yield 200语句后的位置

print(next(g2))
>>
  File "C:/about_gen.py", line 82, in <module>
    print(next(g2))
StopIteration

 

程序发现下一条语句是return,因此抛出StopIteration异常,这样yield 'b'语句永远也不会执行

生成器这个概念一开始很难理解,有点古怪,可是时间久了才知道他的妙用

另外生成器函数是没有办法使用return来返回值

相关文章
相关标签/搜索