python之迭代器和生成器

>>> i = iter('abc')
>>> i.next()
'a'
>>> i.next()
'b'
>>> i.next()
'c'

class Fib(object):
def __init__(self):
self.a, self.b = 0, 1 # 初始化两个计数器a，b

def __iter__(self):
return self # 实例本身就是迭代对象，故返回自己

def next(self):
self.a, self.b = self.b, self.a + self.b # 计算下一个值
if self.a > 100000: # 退出循环的条件
raise StopIteration();
return self.a # 返回下一个值

for n in Fib():
print n

-------------------------------
1
1
2
3
5
...
46368
75025

>>> g = (x*x for x in range(3))
>>> for i in gt :
...    print(i)
0
1
4
>>> for i in g :
...    print(i)

#当你调用这个函数的时候，函数内部的代码并不立马执行, 这个函数只是返回一个生成器对象
def createGenerator():
mylist = range(3)
for i in mylist:
yield i*i

>>> for i in mygenerator:
print i
0
1
4
>>> for i in mygenerator:  #也是只可调用一次
print i