生成器
生成器
一.生成器对象
1.本质
还是内置有__iter__和__next__的迭代器对象
2.区别
迭代器对象是解释器自动提供的
数据类型\文件对象>>>:迭代器对象
生成器对象是程序员编写出来的
代码、关键字>>>:迭代器对象(生成器)
3.创建生成器的基本语法
函数体代码中填写yield关键字
def my_iter():
print('全是科技与狠活儿啊，铁汁')
yield
'''1.函数体代码中如果有yield关键字 那么函数名加括号并不会执行函数体代码，而是会生成一个生成器对象(迭代器对象)'''
res = my_iter()
'''2.使用加括号之后的结果调用__next__才会执行函数体代码'''
res.__next__()
'''3.每次执行完__next__代码都会停在yield位置 下次基于该位置继续往下寻找第二个yield'''
def my_iter():
print('我一勺三花淡奶')
yield 111, 222, 333
print('我一勺十里香')
yield 111, 222, 333
print('千万别整多了了铁汁，整多倒沫子了')
yield 111, 222, 333
print('都是科技与狠活啊铁汁，头发掉光光')
yield 111, 222, 333
res = my_iter()
r1 = res.__next__()
print(r1)
r2 = res.__next__()
print(r2)
r3 = res.__next__()
print(r3)
r4 = res.__next__()
print(r4)
'''4.yield还有点类似于return 可以返回返回值'''
4.练习
自定义生成器对标range功能(一个参数 两个参数 三个参数 迭代器对象)
for i in range(1, 10):
print(i)
# 1.先写两个参数的
# 2.再写一个参数的
# 3.最后写三个参数
def my_range(start_num, end_num=None, step=1):
# 判断end_num是否有值 没有值说明用户只给了一个值 起始数字应该是0 终止位置应该是传的值
if not end_num:
end_num = start_num
start_num = 0
while start_num < end_num:
yield start_num
start_num += step
res = my_range(1, 10).__iter__()
while True:
try:
i = res.__next__() # for i in range(1, 10):print(i)
print(i)
except StopIteration:
break
for i in my_range(100):
print(i)
for i in my_range(1, 10):
print(i)
for i in my_range(10):
print(i)
for i in my_range(100, 50, -1):
print(i)
for i in range(100, 50, -1):
print(i)
5.yield冷门用法
def eat(name, food=None):
print(f'{name}准备用餐')
while True:
food = yield
printf(f'{name}正在吃{food}')
res = eat('jason')
res.__next__()
res.send('汉堡')
'''1.将括号内的数据传给yield前面的变量名 2.再自动调用__next__'''
res.send('薯条')
res.send('鸡翅')
-----------------------------
jason准备用餐
jason正在吃汉堡
jason正在吃薯条
jason正在吃鸡翅
二.生成器表达式
说白了就是生成器表达式
l1 = [i ** 2 for i in range(100)]
print(11)
l1 = (i ** 2 for i in range(100)) # 生成器对象
print(l1) # <generator object <genexpr> at 0x000001DFC07F7E40>
for i in l1
print(i)
面试题(有难度)
def add(n, i): # 普通函数 返回两个数的和求对象
return n + i
def test(): # 生成器
for i in range(4):
yield i
g = test() # 激活生成器
for n in [1, 10]:
g = (add(n, i) for i in g)
"""
第一次for循环 g = (add(n, i) for i in g) 这里相当于给g重新绑定了一个生成器没有运行代码
第二次for循环 g = (add(10, i) for i in (add(10, i) for i in g))
这里就相当于用到for循环去(add(n, i) for i in g)里面取值了，这个时候n是10，生成器给的四个值是0、1、2、3，然后运行add就等于c选项
"""
res = list(g)
print(res)
#A. res=[10,11,12,13]
#B. res=[11,12,13,14]
#C. res=[20,21,22,23]
#D. res=[21,22,23,24]
