Python 中的生成器是什么,yield又是什么作用

最近遇到好多python代码中使用yield的 code,表示的是一个函数是一个生成器。什么意思呢
简答就是说,如果一个函数中使用了yield 定义了一个语句,那么这个函数就不在是一个普通的函数,而是一个生成器,也就是说这个函数是一个可迭代对象

看下面一段代码:

def yield_test(n):  
    for i in range(n):  
        yield callMe(i)  #a
        print("i=",i)    #b  
    print("something else")  
    print("end.")  
  
def callMe(i):  
    return i*i 
    
for i in yield_test(3):
    print(i)

当第一次调用 函数 yield_test ,定义的函数中的code不会像调用其他方法那样依次执行完整个方法,而是在执行到 语句yield (place a 的地方)时,返回后面的函数调用结果。
然后在 for 执行第二次的时候,函数直接从 b 位置开始执行,遇到 yield 语句后,在返回到 for 语句中。依次把for 语句执行完。

试行结果如下:

0
i= 0
1
i= 1
4
i= 2
something else
end.

也可以直接以可迭代对象的方式调用 生成器:yield_test(3).yield_test(3)._next_() (python3.x 以前为next() 函数), 每调用一次返回一个值。

使用生成器的好处是:

  • 生成器会把函数变成一个可迭代对象,每次返回一个值给caller,减少内存消耗(同样的问题不用生成器,可能需要使用list 储存数据然后返回给 caller)
  • 在第二次迭代生成器时,yield 语句之前的变量值会被保存,像是函数被挂住了,第二次会继续执行余下代码,而不是从函数第一行重新开始。
发布了89 篇原创文章 · 获赞 83 · 访问量 3484

猜你喜欢

转载自blog.csdn.net/devin_xin/article/details/105639892
今日推荐