生成器
延迟操作,在需要的时候,调用它,才产生一个元素,节约空间资源,不立即产生全部结果
比如yield
看一个斐波那契数列的例子
#斐波那契
def fib(length):
a,b=0,1
n = 0
while n<length:
yield b #到此停止 相当于return 但是下次调用next时会从上次yield的位置继续执行
n +=1
a,b = b,a+b
return "没有更多内容" #会通过报错来返回
g = fib(10)
try:
print(next(g),next(g),next(g),next(g),next(g))
print(next(g),next(g),next(g),next(g),next(g))
print(next(g),next(g),next(g),next(g),next(g))
except Exception as e:
print(str(e))
生成器是一种特殊的迭代器,是可以迭代的
for i in fib(8):
print(i)
yield也有返回值,可以调用send方法传入返回值
def f():
i=0
while i<10:
t = yield i #t就是send回来的值 i会yield出去
i+=1
print(t)
g = f()
#第一次调用必须先传None
print(g.send(None))
print(g.send("哈哈")) #往生成器里面送值
print(g.send("呵呵"))
迭代器
1、迭代器是一个可以记住遍历的位置的对象。
2、迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
3、迭代器有两个基本的方法:iter() 和 next()。
下面我们来自定义一个迭代器
#################自定义迭代器#############
class WordsIteror:
def __init__(self,string):
self.words = [ w.capitalize() for w in string.split()]
self.index = 0
def __next__(self):
if self.index == len(self.words):
raise StopIteration()
word = self.words[self.index]
self.index +=1
return word
class Words:
def __init__(self,string):
self.string = string
def __iter__(self):
return WordsIteror(self.string)
words = Words("I am Biningo Good morning")
i = iter(words) #__iter__()内置函数返回的值
while True:
try:
print(next(i)) #调用__iter__()返回的对象的__next__()方法
except:
print("Over!")
break
这是一个分割单词的迭代器,调用next()则返回下一个元素,自能前进
下面看一个元祖推导式生成迭代器
arr=(x for x in range(5)) #arr是一个迭代器类型