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 语句之前的变量值会被保存,像是函数被挂住了,第二次会继续执行余下代码,而不是从函数第一行重新开始。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值