Python 实现 itertools.islice

永恒的蓝色梦想

1. class islice:
   
2.     def __init__(self,iterable,start=None,stop=None,step=None,/):
   
3.         if step is None:
   
4.             step=1
   
5.         elif not isinstance(step,int) or step<=0:
   
6.             raise ValueError("Step argument for islice() must be None or a positive integer")
   
7. 
   
8.         if stop is None:
   
9.             stop,start=start,stop
   
10. 
    
11.         if start is None:
    
12.             start=0
    
13.         elif not isinstance(start,int) or start<0:
    
14.             raise ValueError("Start argument for islice() must be None or a non-negative integer")
    
15. 
    
16.         if stop is None:
    
17.             self.__b=None
    
18.         elif isinstance(stop,int)and stop>=0:
    
19.             self.__b=((stop-start-1)//step+1) if start<stop else 0
    
20.         else:
    
21.             raise ValueError("Stop argument for islice() must be None or a non-negative integer")
    
22.         
    
23.         self.__c=step-1
    
24.         self.__a=start
    
25.         self.__it=iter(iterable)
    
26. 
    
27.     def __iter__(self):
    
28.         return self
    
29. 
    
30.     def __next__(self):
    
31.         if self.__b is None:
    
32.             pass
    
33. 
    
34.         elif self.__b:
    
35.             self.__b-=1
    
36. 
    
37.         else:
    
38.             raise StopIteration
    
39. 
    
40.         while self.__a:
    
41.             self.__it.__next__()
    
42.             self.__a-=1
    
43. 
    
44.         self.__a=self.__c
    
45.         return self.__it.__next__()

复制代码

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

_2_

这次的要好好看看了