【Python】 迭代器&生成器

class test(object):
    def __init__(self,num=10):
        self.num = num
        self.li = range(num)
        self.count = 0

    def __iter__(self):    #一般做法就是让__iter__直接返回自身就好了
        return self

    def next(self):
        try:
            if self.count == 0:
                return self.li[0]
            elif self.count <= self.num-1:
                return self.li[self.count]
            else:
                raise StopIteration    #调用迭代器的语句并不知道迭代什么时候才能完成，其往往是按照next给出的逻辑一直迭代下去直到出错。
　　　　　　　　　　　　　　　　　　　　　　　　#想让迭代器在某个边际下停止继续迭代的话，就需要在合适的条件时raise StopIteration来停止迭代。
　　　　　　　　　　　　　　　　　　　　　　　#这个raise出来的并不是会输出到stderr的异常或错误，而是让调用迭代器的语句知道，可以不用再调用next方法来获取下一个值了
        finally:
            self.count+=2   #这里还有一个比较tricky的，和迭代器本身没啥关系的点。。之前我在写count+=2的时候发现，无论写哪里都不太好。
　　　　　　　　　　　　　　　　　#因为按照现有语句的逻辑，最好是在next方法返回值后count能再加一次2，这样下次进next万一超过了num-1的限制就可以不用再return。
　　　　　　　　　　　　　　　　　#　但是一般而言，return之后的语句在本次调用函数的过程中时肯定不会再被执行了，除了try/finally结构。
　　　　　　　　　　　　　　　　　#　就是说，finally中的语句，即便是在return执行了之后还是会被执行的。这点比较难想到！

if __name__ == ‘__main__‘:
    test = test()
    for item in test:
        print item

def flatten(nested_thing):
    for element in netsed_thing:
        if not isinstance(element,list):
            yield element
        else:
            for subelement in flatten(element):
                yield element

nested = [[1,2],[3,4],5,[[6,7],8]]
result = list(flatten(nested))
print result

#结果是[1,2,3,4,5,6,7,8]