[ python ] 迭代器和生成器

 列表生成式

  列表生成式是 python 内置的非常简单却强大的可以用来创建list的生成式。

有这样一个需求:
    l1 = [1,2,3,4,5,6,7,8,9]    循环每个元素加1

l1 = [1,2,3,4,5,6,7,8,9]

# 使用列表生成式将列表中的各个元素加1
list_num = [x+1 for x in l1]	# 这里就是一个列表生成式
print(list_num)

列表生成式可以用一行语句代替循环生成上面的list

 迭代器

    迭代器表面上看是一个数据流对象或容器,当使用其中的数据时,每次从数据流中取出一个数据,直到数据被取完,而且数据不会被重复使用。
    从代码的角度来看,迭代器是实现了迭代器协议方法的对象和类。
    迭代器协议方法主要是两个:
        __iter__(): 该方法返回对象本身,它是for语言使用迭代器的要求,只要函数含有__iter__就可以说这个函数是可迭代的;
        __next__(): 该方法用于返回容器中下一个元素或数据。当容器中的数据用尽时,应该引发 StopIteration 异常一个类只要具有这两种方法,就可以称之为 迭代器,也可以说是可迭代的。

s1 = 'hello world'
print('__iter__' in dir(s1))
print('__next__' in dir(s1))

# 字符串s1包含 __iter__ 方法且不包含__next__方法,所以字符串 s1 只是一个可迭代的类型,并不是一个迭代器

# 执行结果:
# True
# False
s1 = 'hello world'
s2 = s1.__iter__()  # 将可迭代类型转换为 迭代器 使用 __iter__()
print('__iter__' in dir(s2))
print('__next__' in dir(s2))

# 使用 __iter__()方法将 s1 字符串转换为迭代器,迭代器既有__iter__方法,又有 __next__方法
# 执行结果:
# True
# True

自定义一个迭代器必须满足:
    类中必须要定义 __iter__ 方法和 __next__方法

# 自定义一个迭代器:

class My_iterator(object):
    def __init__(self, x, max):
        self.mul, self.x = x, x
        self.xmax = max

    def __iter__(self):
        return self

    def __next__(self):
        if self.x and self.x !=1:
            self.mul = self.mul + self.x
            if self.mul < self.xmax:
                return self.mul
            else:
                raise StopIteration

if __name__ == '__main__':
    myite1 = My_iterator(2, 100)
    for i in myite1:
        print(i)

 生成器

  使用生成器可以生成一个值的序列用于迭代,并且这个值的序列不是一次生成的,而是使用一个,再生成一个,可以使得程序节约大量内存。
    生成器对象是通过yield关键字定义的函数对象,因此生成器也是一个函数。生成器用于生成一个值的序列,以便在迭代中使用。

自定义生成器:

def foo():
    print('1111')
    yield 1
    print('22222')
    yield 2

f = foo()
print('type f:', type(f))
print(f.__next__())
print(f.__next__())

# 执行结果:
# type f: <class 'generator'>
# 1111
# 1
# 22222
# 2

执行流程如下图:

需要注意的是 yield 是生成器中的关键字,生成器在实例化的时候并不会立即执行,而是等待调用其__next__()方法之后才开始运行,当再次调用__next__()方法会在第一次yield返回值的最后开始执行,不断的调用__next__()方法,直到最终返回 StopIteration 异常为止。

猜你喜欢

转载自www.cnblogs.com/hukey/p/9698150.html