python 设计模式-迭代器模式

迭代器是一种很常用的设计模式，python 在语言层面也对迭代器进行了友好的支持，我前面有一篇介绍迭代器和可迭代对象的文章，初学者可以先了解下概念，这里主要通过代码示例进行说明。

#!/usr/bin/env python #coding: utf-8 class ProductLst: def __init__(self): self._product_lst = [] self._idx = -1 def add(self, product): self._product_lst.append(product) def __iter__(self): self._idx = -1 return self def __next__(self): self._idx += 1 if self._idx == len(self._product_lst): raise StopIteration return self._product_lst[self._idx] if __name__ == '__main__': ps = ProductLst() ps.add('pa') ps.add('pb') ps.add('pc') for p in ps: print(p)

输出：

pa pb pc

我们只要在需要迭代的对象上定义__next__方法，在for循环的时候会自动进行迭代遍历。