实现了一下itertools.tee

永恒的蓝色梦想

闲得无事，写了个函数，实现了和 itertools.tee 类似的功能

这个函数有什么用？

这个 tee 使用链表实现。

代码：

1. class node:
   
2.     def __init__(self,value):
   
3.         self.value = value
   
4.         self.next = None
   
5. 
   
6. 
   
7. class _tee:
   
8.     def __init__(self,iter,node):
   
9.         self.iter = iter
   
10.         self.node = node
    
11. 
    
12.     def __iter__(self):
    
13.         return self
    
14. 
    
15.     def __next__(self):
    
16.         if self.node.next is None:
    
17.             try:
    
18.                 self.node.next = self.node = node(self.iter.__next__())
    
19. 
    
20.             except StopIteration as err:
    
21.                 self.node.next = NotImplemented
    
22.                 raise err
    
23.         
    
24.         elif self.node.next is NotImplemented:
    
25.             raise StopIteration
    
26. 
    
27.         else:
    
28.             self.node = self.node.next
    
29. 
    
30.         return self.node.value
    
31. 
    
32. 
    
33. def tee(iterable,n=2,/):
    
34.     temp = []
    
35.     iterator = iter(iterable)
    
36.     init = node(None)
    
37. 
    
38.     while n:
    
39.         temp.append(_tee(iterator,init))
    
40.         n-=1
    
41. 
    
42.     return tuple(temp)

复制代码

永恒的蓝色梦想

二楼

zltzlt

1. def tee(iterable, n=2, /):
   
2.         return tuple(iterable for _ in range(n))

复制代码

哈哈哈

一个账号

你的鱼币肿么变成负数了

永恒的蓝色梦想

zltzlt 发表于 2020-3-31 17:31
哈哈哈

这个不行的，

1. >>> from itertools import tee
   
2. >>> a,b,c=tee([1,2,3,4].__iter__(),3)
   
3. >>> list(a)
   
4. [1, 2, 3, 4]
   
5. >>> list(b)
   
6. [1, 2, 3, 4]
   
7. >>> list(c)
   
8. [1, 2, 3, 4]
   
9. >>> def tee(iterable, n=2, /):
   
10.         return tuple(iterable for _ in range(n))
    
11. 
    
12. >>> a,b,c=tee([1,2,3,4].__iter__(),3)
    
13. >>> list(a)
    
14. [1, 2, 3, 4]
    
15. >>> list(b)
    
16. []
    
17. >>> list(c)
    
18. []

复制代码

tee 需要复制出来的迭代器都能完整迭代一次