自定义迭代器iterator

阅读更多

python中迭代器的定义方式主要有两种，一种是通过类实现__iter__()方法，其返回一个实现了__next__()方法的object，如果该类自身实现了__next__()，则只需直接返回self即可，如下是一个拥有逆置功能的类：

#创建过程：
class Reverse:
    def __init__(self, data):
        self.data = data
        self.index = len(data)

    def __iter__(self):
        return self

    def __next__(self):
        if self.index == 0:
            raise StopIteration
        self.index -= 1
        return self.data[self.index]

#调用结果如下：
d = 'abcdefghi'
r = Reverse(d)
for ch in r:
    print(ch)

#输出结果：
i
h
g
f
e
d
c
b
a

另一种方式是利用生成器generator，通过关键字yield，简单实用，主要用于函数中，如下：

#创建过程：
def reverse(data):
    for i in range(len(data), 0, -1):
        yield data[i-1]

#调用过程：
d = 'abcdefg'
for ch in reverse(d):
    print(ch)

#输出结果：
g
f
e
d
c
b
a

从定义方式可看出，两者的区别主要是应用对象不同：前者用在类中，而后者用在函数或方法中。