Python 实现 head()

永恒的蓝色梦想

1. class head:
   
2.     def __init__(self,iterable,n,/):
   
3.         if isinstance(n,int) and n>=0:
   
4.             self.__n=n
   
5.         else:
   
6.             raise TypeError(f"'{n.__class__.__name__}' object cannot be interpreted as an integer")
   
7. 
   
8.         self.__it=iter(iterable)
   
9. 
   
10.     def __iter__(self):
    
11.         return self
    
12. 
    
13.     def __next__(self):
    
14.         if self.__n:
    
15.             self.__n-=1
    
16.             return self.__it.__next__()
    
17. 
    
18.         raise StopIteration

复制代码