少说话多写代码之Python学习051——类的成员（实现迭代器）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/yysyangyangyangshan/article/details/84575639

在Python类中，实现迭代器，要定义这个方法，__iter__。这个方法返回一个迭代器，其实质是带有一个next方法的对象。为形象起见，我们定义一个斐波那契数列类。如下，

class MyFibs:
    def __init__(self):
        self.before=0
        self.after=1
    def __next__(self):
        self.before,self.after = self.after,self.before + self.after
        return self.before
    def __iter__(self):
        return self

调用

fib = MyFibs()
for item in fib:
   if item<5:
       print(item)
   else:
       break

打印出5以前的斐波那契数列，

1
1
2
3

这里需要注意，本文中使用的是Python3.0，需定义__next__函数，如果定义next，会报如下异常：
typeError: iter() returned non-iterator of type比如这样写就报错了。   

class MyFibs2:
    def __init__(self):
        self.before=0
        self.after=1
    def next(self):
        self.before,self.after = self.after,self.before + self.after
        return self.before
    def __iter__(self):
        return self

fib2 = MyFibs2()
for item in fib2:
   if item<5:
       print(item)
   else:
       break

我们尝试用迭代器实现序列。实现__next__和 __iter__。

class UseMyIterator:
    num=0
    def __next__(self):
        self.num=self.num+1
        if self.num>20:raise StopIteration
        return  self.num
    def __iter__(self):
        return self

调用

t = UseMyIterator()
print(list(t))

输出20个序列

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]

工程文件下载：https://download.csdn.net/download/yysyangyangyangshan/10811830