论坛 › Python交流 › Python 实现 itertools.islice

永恒的蓝色梦想 发表于 2020-4-10 08:30:39

Python 实现 itertools.islice

本帖最后由 永恒的蓝色梦想 于 2020-4-11 09:44 编辑

class islice:
    def __init__(self,iterable,start=None,stop=None,step=None,/):
      if step is None:
            step=1
      elif not isinstance(step,int) or step<=0:
            raise ValueError("Step argument for islice() must be None or a positive integer")

      if stop is None:
            stop,start=start,stop

      if start is None:
            start=0
      elif not isinstance(start,int) or start<0:
            raise ValueError("Start argument for islice() must be None or a non-negative integer")

      if stop is None:
            self.__b=None
      elif isinstance(stop,int)and stop>=0:
            self.__b=((stop-start-1)//step+1) if start<stop else 0
      else:
            raise ValueError("Stop argument for islice() must be None or a non-negative integer")
      
      self.__c=step-1
      self.__a=start
      self.__it=iter(iterable)

    def __iter__(self):
      return self

    def __next__(self):
      if self.__b is None:
            pass

      elif self.__b:
            self.__b-=1

      else:
            raise StopIteration

      while self.__a:
            self.__it.__next__()
            self.__a-=1

      self.__a=self.__c
      return self.__it.__next__()如果代码有问题，欢迎在评论区指正！{:10_323:}

_2_ 发表于 2020-4-10 10:07:20

这次的要好好看看了

查看完整版本: Python 实现 itertools.islice