Python 实现 itertools.cycle

zltzlt

Python 实现 itertools.cycle

1. class cycle:
   
2.     def __init__(self, iterable):
   
3.         self.__it = iter(iterable)
   
4.         self.__res = []
   
5.         self.__flag = False    # 迭代器是否已无元素可取
   
6. 
   
7.     def __iter__(self):
   
8.         return self
   
9. 
   
10.     def __next__(self):
    
11.         if self.__flag:
    
12.             try:
    
13.                 return next(self.__it)
    
14.             except StopIteration:
    
15.                 self.__it = iter(self.__res)
    
16.                 return next(self.__it)
    
17.         else:
    
18.             try:
    
19.                 val = next(self.__it)
    
20.                 self.__res.append(val)
    
21.                 return val
    
22.             except StopIteration:
    
23.                 self.__flag = True
    
24.                 self.__it = iter(self.__res)
    
25.                 return next(self.__it)

复制代码

如果代码有错误，欢迎在下方留言指出

永恒的蓝色梦想

我也整一个

1. def cycle(iterable, /):
   
2.     save=[]
   
3. 
   
4.     for i in iterable:
   
5.         yield i
   
6.         save.append(i)
   
7. 
   
8.     if save:
   
9.         while True:
   
10.             yield from save

复制代码