生成器
在一个函数中经常使用 return
关键字返回数据,但是有时候会使用 yield
关键字返回数据。使用 yield
关键字的函数返回的是一个生成器(generator)对象
,生成器对象是一种可迭代对象。
def square(num):
"""定义参数函数"""
n_list = []
for i in range(1, num + 1):
n_list.append(i * i)
return n_list
for i in square(5):
print(i)
yield
更简单的方法
def square1(num):
for i in range(1, num + 1):
yield i * i
for i in square1(5):
print(i)
yield关键字返回平方数,不在需要return关键字了。
生成器对象是一种可迭代对象,可以迭代对象通过__next__() 方法获得元素,for循环就是隐式地调用了生成器的__next__() 方法。
生成器函数通过yeild返回数据,与return不同的是,return语句一次返回所有数据,函数调用结束;而yiled语句只返回一个元素数据,函数调用不会结束,只是暂停,直到__next__() 方法被调用,程序继续执行yield语句之后的语句代码。
生成器特别适合用于遍历一些大序列对象,它无需将对象的所有元素都加载到内存后才开始进行操作,仅在迭代某个元素时才会将该元素加载到内存。