迭代器和生成器

一、迭代器

1、定义

  迭代器(Iterator)是一种对象,它能够用来遍历标准模板库容器中的部分或全部元素,每个迭代器对象代表容器中的确定的地址。

满足什么样的条件才能成为迭代器呢?

  • 必须遵从迭代器协议 

迭代器协议:对象必须提供一个next方法,执行该方法要么返回迭代中的下一项,要么就引起一个StopIteration异常,以终止迭代 。

满足什么样条件的对象才能提供一个next方法呢?

  • 对象内部定义了__iter__()以及__next__()方法
class Fibonacci(object):
    def __init__(self,m):
        self.m=m

    def __iter__(self):
        return self   #自身是一个迭代器

    def __next__(self):
        n, a, b = 0, 0, 1
        while n < self.m:
            print(b)
            a, b = b, a + b
            n += 1
f=Fibonacci(5) #f此时就是一个迭代器
for  i in f:
    print(i)

2、可迭代对象

  可迭代对象内部仅实现了__iter()__方法,可迭代对象可以通过._ iter _()方法转成迭代器。像listtupledictsetstr等都是直接可以使用for循环的对象都是可迭代对象。

list=[1,2,3]
#手动实现__iter__()方法
ilist=list.__iter__() 
print(ilist.__next__())#1
print(ilist.__next__())#2
print(ilist.__next__())#3

  而在for循环内部也是基于这样的实现方式,不过在for循环内部还进行了异常处理。

#####for循环的实质#######
list=[1,2,3]
#手动实现__iter__()方法
ilist=list.__iter__()
while True:
    try:
        print(ilist.__next__())
    except StopIteration as e:
        break

3、生成器

  可以理解为一种数据类型,这种数据类型自动实现了迭代器协议,即内部自动实现了__iter__()和__next__()方法,所以生成器不仅是可迭代对象而且还是迭代器对象。那么所有可迭代对象拥有的特性,它都拥有,比如for循环。

  如何构造一个生成器呢?

  在一个函数中通过yield关键字自动创建一个生成器函数,yield会自动构建好__next__()和__iter__()方法。

def test():
    yield 1
    yield 2
    yield 3

t=test()
print(type(t))#<class 'generator'>
print(t.__next__())#1
print(t.__next__())#2
print(t.__next__())#3

  进行for循环

for i in test():
    print(i)  #1,2,3

  所以生成器就是迭代器的一种,是构造迭代器的工具。

def Fibanocci(m):
    n,a,b = 0,0,1
    while n < m:
        yield b
        a,b=b,a+b
        n+=1
f = Fibanocci(5)   #f为一个迭代器对象,可以将其当作xrange来对待
while True:
    try:
        print(f.__next__())
    except StopIteration as e:        
      break

  在类中也是可以实现迭代器的:

class Fibonacci(object):
    def __init__(self,m):
        self.m=m

    def __iter__(self):
        n, a, b = 0, 0, 1
        while n < self.m:
            yield b
            a, b = b, a + b
            n += 1
f=Fibonacci(5) #f此时就是一个迭代器
for i in f:
    print(i)

  如何判断一个对象是可迭代对象呢?可以使用isinstance()进行判断。

from collections import Iterable
print(isinstance([], Iterable))#True
print(isinstance({}, Iterable))#True

  如何判断是否是生成器对象?可以使用isgenerator()进行判断

from inspect import isgenerator
def test():
    yield 1

t=test()
print(isgenerator(t))#True

  详情查看:https://docs.python.org/3/library/inspect.html#inspect.isgeneratorfunction

  如何判断是迭代器对象呢?可以使用isinstance()进行判断

from collections import Iterator
a="abc"
b=iter(a)
print(type(b))
print(isinstance(b,Iterator))#True

总结:

  迭代器对象内部实现了__iter__()和__next__()方法

  可迭代对象仅实现__iter__()方法,必须依靠对象调用__iter__()方法(iter(o))才能成为迭代器对象

  迭代器是可迭代对象,但可迭代对象不一定是迭代器。

  生成器既是可迭代对象,又是迭代器对象,相当于xrange。

  

原文地址:https://www.cnblogs.com/shenjianping/p/11042098.html