1.生成器函数初识
生成器的本质还是一个迭代器,而生成器函数本质上就是我们自己写的函数
生成器的应用场景是:想使用大量数据,但又不想占用很大的内存
首先我们来对比一个普通函数和生成器函数:
# 1.普通函数
def generator():
print(1)
return 'a'
ret = generator()
print(ret)
# 2.生成器函数
def generator():
print(1)
yield 'a'
ret = generator()
print(ret)
我们可以看到,生成器函数中的返回使用yield关键字返回,而不是return关键字。
因此我们定义如下:
含有yield关键字的函数都是生成器函数
yield的特点是:
yield关键字只能用于函数内部,不能用于函数外部
yield与return只能共存一个
yield与return的区别是:
return会直接结束掉函数而yield不会(可以理解yield关键字为暂停和播放);
在遇到yield关键字处函数会返回值然后暂停,下次调用函数时,会从暂停的下一条语句处开始执行
但是yield与return都会返回值
2.生成器函数使用
1.基本使用
def generator():
print(1)
yield 'a'
# 遇到yield关键字以后会得到一个生成器作为返回值,且不执行函数内的语句
ret = generator()
print(ret)
print('__iter__' in dir(ret))
print('__next__' in dir(ret))
# 说明ret是一个迭代器,即生成器也是一个迭代器
print(ret.__next__())
输出结果如下:
2.生成器函数的取值方法
顺序取值
def generator():
for i in range(10)<