Python 迭代器和生成器

浏览数：19 / 时间：2015年06月08日

一直以为 Python 的生成器是指列表生成，好吧，我读书少。

其实呢，生成器是使用yield 返回实现了迭代器协议的generator 对象。如下：

class Data(object):
    def __init__(self, *args):
        self._data = list(args)
    def __iter__(self):
        for x in self._data:
            yield x
    d = Data(1,2,3)
    >>> d.__iter__()
    <generator object __iter__ at 0x7f8556d4cdc8>

在python2 中使用d.next() 可以返回 d 的下一个值，
在 python3 中需要使用 d.next() 返回 d 的生成器的下一个值，生成器可用于协程够造。

关于迭代器
根据迭代器协议, 需要实现iter() 和 next() 两个方法, 前者返回迭代器对象，后者返回数据，直到引发StopIteration 异常结束。

class Data(object):
def __init__(self, *args):
    self.data = list(args)
    self.index = 0
def __iter__(self):
    return self
def __next__(self):
    if self.index < len(self.data):
        d =  self.data[self.index]
        self.index +=1
    else:
        raise StopIteration()
    return d

郑重声明：本站内容如果来自互联网及其他传播媒体，其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享，并不代表本站赞同其观点和对其真实性负责，也不构成任何其他建议。

Python 迭代器和生成器