Python迭代器与生成器

佚名 6年前 (2019-04-16) 随笔 1025人围观抢沙发百度已收录

Python 迭代器与生成器
1.迭代器
迭代器是集合元素的一种方式
迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退
迭代器有两个基本的方法：iter() 和 next()
字符串，列表或元组对象都可用于创建迭代器

案例描述：
list=[1,2,3,4]
it = iter(list)    # 创建迭代器对象
print (next(it))   # 输出迭代器的下一个元素
print (next(it))   # 输出end+1 抛出异常

list=[1,2,3,4]
it = iter(list)    # 创建迭代器对象
for x in it:       # for语句进行遍历
　　print (x, end=" ")


import sys         # 引入 sys 模块
list=[1,2,3,4]
it = iter(list)    # 创建迭代器对象
while True:        # while循环进行遍历
　　try:
　　　　print (next(it))
　　except StopIteration: # StopIteration 异常用于标识迭代的完成，防止出现无限循环的情况
   　　　 sys.exit()

2.生成器yield
在 Python 中，使用了 yield 的函数被称为生成器（generator）
生成器是一个返回迭代器的函数，只能用于迭代操作

import sys
def fibonacci(n): # 生成器函数 - 斐波那契
　　a, b, counter = 0, 1, 0
　　while True:
　　　　if (counter > n):
　　　　　　return
　　　　yield a
   　　　a, b = b, a + b
   　　    counter += 1

f = fibonacci(10) # f 是一个迭代器，由生成器返回生成

while True:
  try:
   print (next(f), end=" ")
  except StopIteration:
   sys.exit()

在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行
每次调用这个函数并使用迭代器进行循环的时候一个一个的得到每个 list 元素而不是直接得到一个完整的 list 来节省内存