Python 实现 itertools.chain

永恒的蓝色梦想

两种实现

1. def chain(*iterables):
   
2.     for iterable in iterables:
   
3.         yield from iterable

复制代码

1. class chain:
   
2.     def __init__(self,*iterables):
   
3.         self.__iterables=map(iter,iterables)
   
4.         self.__doing=self.__iterables.__next__()
   
5. 
   
6.     def __iter__(self):
   
7.         return self
   
8. 
   
9.     def __next__(self):
   
10.         while True:
    
11.             try:
    
12.                 return next(self.__doing)
    
13.             
    
14.             except StopIteration:
    
15.                 self.__doing=self.__iterables.__next__()
    
16. 
    
17.     @classmethod
    
18.     def from_iterable(cls,iterable,/):
    
19.         return cls(*iterable)

复制代码

如果代码有错误，欢迎在评论区指正

qiuyouzhi

用yield from会提高代码运行速度吗？

_2_

qiuyouzhi 发表于 2020-4-3 10:10
用yield from会提高代码运行速度吗？

自己用 IPython 试试

永恒的蓝色梦想

qiuyouzhi 发表于 2020-4-3 10:10
用yield from会提高代码运行速度吗？

不知道，写起来简单倒是真的