解决生成器不能复用-----生成器列表

Python中的生成器 有函数形式和列表生成器,但是他们都有一个缺点是不能复用,一次用完便不能再用,但是有时候我们需要复用结果,自然想到有没有一种叫做生成器列表的东西,第一次用的时候用生成器创建数据并将数据存起来,之后直接访问一保存的数据不再使用生成器,代码如下:

import collections

class gen_list(collections.abc.Sequence):
    '''借助原生的Iteration'''
    def __init__(self,generator):
        self.__generator=generator
        self.object_list=[]
        self.not_finish=True

    def __getitem__(self, item):
        if self.not_finish:
            try:
                i=next(self.__generator)
                self.object_list.append(i)
            except StopIteration:
                self.not_finish=False
        return self.object_list[item]

    def __len__(self):
        _len=self.object_list.__len__()
        return 1 if _len is 0 else _len

测试:

if __name__ == '__main__':
    def gen():
        for i in range(10):
            yield i
    a=gen_list(gen())
    for i in a:
        print('A',i)
    print('\n')
    for i in a:
        print('B',i)

结果:


A 0
A 1
A 2
A 3
A 4
A 5
A 6
A 7
A 8
A 9


B 0
B 1
B 2
B 3
B 4
B 5
B 6
B 7
B 8
B 9

Process finished with exit code 0

猜你喜欢

转载自blog.csdn.net/a1053904672/article/details/87874830
今日推荐