迭代器和生成器

一、迭代器

什么是迭代器协议:对象必须提供一个next()方法,执行该方法返回迭代中的元素,没有则抛Stopiteration错误

遵循迭代器协议的对象就是可迭代对象                        (如何实现可迭代对象:对象内部实现一个__iter__()方法)

能被for循环的对象,全部都遵循迭代器协议(内部都是调用__iter__()方法)  for 循环就是基于迭代器协议来工作的,for循环做了两件事,第一件事就是将对象变成可迭代对象,并且捕获异常

python中的字符串,列表,元组,字典,集合,文件 本身并没有next()方法,在用for循环遍历的时候,

实际是该数据类型调用了内部的__iter__()方法,将数据类型变成了可迭代对象,该可迭代对象具有__next__()的方法

例如:

a = 'hello'
iter_test = a.__iter__()
print(iter_test.__next__())
print(iter_test.__next__())
print(iter_test.__next__())
print(iter_test.__next__())
print(iter_test.__next__())
print(iter_test.__next__())

>>>

h
e
l
l
o

StopIteration

 

二、生成器就是可迭代对象,自动实现了迭代器协议,不用调用__iter__()方法就有__next__()方法

两种生成器表达式:

函数中只要有yield就是生成器

生成器表达式:列表生成式 (i for i in range(10))

#############  yield 操作

yield特性:

yield 1  相当于函数的 return 1   ,可以返回值

yield还可以接收值, 如下例:(next方法可以执行生成器以外,send方法也可以,并且还可以传递值,赋值给当前位置的上一个yield)

def foo():
  print('a')
  ffff = yield 1
  print('b', ffff)
  yield 2

res = foo()
print(res.__next__())
print(res.send('send'))

 

posted @ 2018-06-22 02:33  BensonChang  阅读(140)  评论(0编辑  收藏  举报