Python 迭代器、生成器、可迭代对象

迭代器

#迭代器定义：
#类中得有__iter__和__next__两个方法
#__iter__方法放回对象本身，即：self(是迭代器对象)
#__next__方法，返回下一个数据，若是没有了，要抛出一个 StopIteration异常

class Li(object):
     def __init__(self):
         self.counter=0

     def __iter__(self):
         return self
     def __next__(self):
         self.counter += 1
         if self.counter==5:
             raise StopIteration()
         return self.counter
obj1=Li()

#测试返回迭代器对象本身，可以循环
#结果为：1 2 3 4
v1=obj1.__iter__()
for item in v1:  #这里v1换成obj1结果一样
    print(item)
#for循环首先执行迭代器对象的__iter__方法，获取到其返回值（self）
#然后for循环反复执行其__next__方法，每执行一次就赋值给item
#for循环依赖于迭代器



#测试obj1.__next__(),写几个执行几次，直到报错
#与obj1.__next__()等同的是自带的next(obj1)

print(dir(v1))  #dir查其方法

生成器

#生成器(特殊的迭代器)
#编写方式和表现形式不同罢了
def func():
    yield 1
    yield [1, 2, 3, 4, ]
    yield {'name':'Kant','age':18}
#创建生成器对象(内部是根据生成器类generator创造出对象)，其内部也有__iter__和__next__两个方法
obj=func() #生成器对象

#print(next(obj))
for i in obj:
    print(i)  #生成上面完全一样的东西
#也是基于迭代器去取值
print(dir(obj))

可迭代对象

#如果一个类中有__iter__方法且返回一个迭代器对象;则我们称这个类创建的对象为可迭代对象
#列表  字典  都是可迭代对象

#
class Li(object):
    def __init__(self):
        self.counter = 0

    def __iter__(self):
        return self

    def __next__(self):
        self.counter += 1
        if self.counter == 5:
            raise StopIteration()
        return self.counter




class Foo(object):
    def __iter__(self):
        obj=Li()
        return obj  #return一个迭代器对象obj
        #也可以直接写成return Li()

obj=Foo()  # 创建可迭代对象obj(有__iter__方法,无__next__方法)
print(dir(obj))
for i in obj:
    print(i)
#循环可迭代对象的时候，内部先执行obj.__iter__并且获取迭代器对象
#然后不断循环迭代器对象的__next__方法

#判断是可迭代对象还是迭代器
from collections.abc import Iterator,Iterable
#Iterator迭代器（有__iter__和__next__两个方法）
#Iterable是否可迭代（有__iter__并且返回一个迭代器对象）
v1=[1,2,3,4,]     #为可迭代对象
print(isinstance(v1,Iterator)) #False,不是迭代器
print(isinstance(v1,Iterable)) #True,可以迭代
obj66=Li()    #为迭代器对象
print(isinstance(obj66,Iterable)) #True,因为它也返回自身(迭代器)

#判断一个东西是否为可迭代对象
def judge(obj):
    if isinstance(obj,Iterable):
        if isinstance(obj,Iterator):
            print('这是一个迭代器对象')
        else:
            print('这是一个可迭代对象')
    else:
        print('这个对象不可迭代')

judge(v1)
#这是一个可迭代对象