[Python知识库]python迭代器和生成器

迭代器

• 运行步骤：
  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))