在Queue中实现__iter__(self)

ABC23

如何在Python的队列结构中实现迭代器？
=========================
class Queue(object):
        def __init__(self):
                self.queue = []                # 用列表模拟队列（封装成一个类）
        def __iter__(self):
                return self
        def __next__(self):
                # 关键是这个怎么写。。
====================

第四时空

这样？

1. class Queue(object):
   
2.     def __init__(self):
   
3.         self.queue = [1, 2, 3]
   
4. 
   
5.     def __iter__(self):
   
6.         return iter(self.queue)
   
7. 
   
8. 
   
9. if __name__ == '__main__':
   
10.     for i in Queue():
    
11.         print(i)

复制代码

ABC23

我没说清楚。
不要借助list的iter。怎么写？

第四时空

ABC23 发表于 2018-4-11 18:03
我没说清楚。
不要借助list的iter。怎么写？

1. class Queue(object):
   
2.     def __init__(self):
   
3.         self.queue = [1, 2, 3]
   
4.         self.__index = 0
   
5. 
   
6.     def __iter__(self):
   
7.         return self
   
8. 
   
9.     def __next__(self):
   
10.         if self.__index == len(self.queue):
    
11.             self.__index = 0
    
12.             raise StopIteration
    
13.         elem = self.queue[self.__index]
    
14.         self.__index += 1
    
15.         return elem

复制代码

1. class Queue(object):
   
2.     def __init__(self):
   
3.         self.queue = [1, 2, 3]
   
4. 
   
5.     def __getitem__(self, item):
   
6.         return self.queue[item]

复制代码

1. class Queue(object):
   
2.     def __init__(self):
   
3.         self.queue = [1, 2, 3]
   
4. 
   
5.     def __iter__(self):
   
6.         for i in self.queue:
   
7.             yield i

复制代码

ABC23

为什么要写__index，而不是index?

第四时空

ABC23 发表于 2018-4-11 18:32
为什么要写__index，而不是index?

写index也行，写__index的原因是出于一些安全考虑，index理应是我们类内部维护的一个值，不想被外部随意修改当前的index的值。导致迭代不稳定。

ABC23

嗯。谢了。