Python 实现 itertools.takewhile

永恒的蓝色梦想

这里给出两种实现

1. class takewhile:
   
2.     def __init__(self,predicate,iterable,/):
   
3.         self.__predicate=predicate
   
4.         self.__iterable=iter(iterable)
   
5. 
   
6.     def __iter__(self):
   
7.         return self
   
8. 
   
9.     def __next__(self):
   
10.         temp=next(self.__iterable)
    
11. 
    
12.         if self.__predicate(temp):
    
13.             raise StopIteration
    
14. 
    
15.         else:
    
16.             return temp

复制代码

1. def takewhile(predicate, iterable, /):
   
2.     for i in iterable:
   
3.         if predicate(i):
   
4.             break
   
5.         else:
   
6.             yield i

复制代码

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