迭代器
- 运行步骤:
1.可迭代对象通过__iter__拿到迭代器
2.执行迭代器的__next__拿到元素,反复执行,直到取到最后一个元素
lst = ["Baidu", "Google"]
it = lst.__iter__()
print(it.__next__()) # Baidu
print(it.__next__()) # Google
print(it.__next__()) # 报错:StopIteration
-
特点:
1.节省内存
2.惰性机制(必须访问__next__才能拿到数据)
3.不能反复,只能向下访问 -
for循环内部工作机制
lst = ["Baidu", "Google", "JD"]
# 拿到迭代器
it = lst.__iter__()
while True:
try:
# 获取数据
print(it.__next__())
except StopIteration:
break
生成器
生成器的本质就是迭代器,将函数中的return换成yield就成为生成器函数
– 里面有yield
– 生成器函数在执行时,实际上是创建一个生成器出来
– 须使用__next__()来执行代码,到下一个yield结束
– yield相当于返回,让一个函数分段执行
– 当执行到最后一个yield时,再次执行__next__()会报错StopIteration
- 生成器的作用
节省内存
def order():
lst = []
for i in range(1000):
lst.append(f"第{i}块糖")
if len(lst) == 50:
# 当取满50块糖时,停止运行
yield lst
# 下一次再需要时,再次执行__next__()
lst = []
lst = order()
# __next__()到哪,指针就到哪,下一次执行从指针指向的值开始
print(lst.__next__())
print(lst.__next__())
- 生成器表达式
基本语法:(结果 for循环 if)
gen = (result for result in range(10))
# 将生成器中的数据全部拿出
# 1.for循环全部拿出生成器中的数据
for item in gen:
print(item)
# 2.使用list tuple set
print(list(gen))
- 练习
def func():
print(111)
yield 222
# 创建了一个生成器
g = func()
# 创建生成器g1
g1 = (i for i in g)
# 创建生成器g2
g2 = (i for i in g1)
# 使用list将生成器g中的数据拿空了,g1和g2中数据为空
print(list(g))
print(list(g1))
print(list(g2))