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

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值