Python 实现 itertools.zip_longest

永恒的蓝色梦想

两种实现

1. def zip_longest(*iterables, fillvalue=None):
   
2.     iterators=tuple(iter(i) for i in iterables)
   
3. 
   
4.     while True:
   
5.         temp=[]
   
6.         flag=True
   
7. 
   
8.         for i in iterators:
   
9.             try:
   
10.                 temp.append(next(i))
    
11.                 flag=False
    
12. 
    
13.             except StopIteration:
    
14.                 temp.append(fillvalue)
    
15. 
    
16.         if flag:
    
17.             break
    
18. 
    
19.         yield tuple(temp)

复制代码

1. class zip_longest:
   
2.     def __init__(self,*iterables,fillvalue=None):
   
3.         self.__iterators=tuple(iter(i) for i in iterables)
   
4.         self.__filevalue=fillvalue
   
5. 
   
6.     def __iter__(self):
   
7.         return self
   
8. 
   
9.     def __next__(self):
   
10.         temp=[]
    
11.         flag=True
    
12. 
    
13.         for i in self.__iterators:
    
14.             try:
    
15.                 temp.append(next(i))
    
16.                 flag=False
    
17.             
    
18.             except StopIteration:
    
19.                 temp.append(self.__filevalue)
    
20. 
    
21.         if flag:
    
22.             raise StopIteration
    
23. 
    
24.         return tuple(temp)

复制代码

如果代码有问题，欢迎在评论区留言指出！

_2_

渣渣表示没用过