先介绍一下迭代器的概念:
迭代器从对象集合的第一个元素开始访问,直到所有的元素被访问完,可以记住遍历的位置,只会往前不会后退。
迭代器实现了_iter_方法和_next_方法,可以通过_next_方法不断返回下一个值的对象称为迭代器。列表、元组、字符串、字典实现了_iter_方法,但是没有实现_next_方法,因此不能成为迭代器。
v=['train','bus','car']
vehicle=iter(v)#通过iter()函数获得一个迭代器
v._next_()#结果为train
判断一个对象是否为迭代器:
isinstance()函数
isinstance(123,Iterator)
False
生成器是一种特殊的迭代器,只需要在函数中使用yield关键字逐一返回元素。Python中用yield函数(生成器函数)返回生成器对象。
#打印输出斐波那契额数列的前n个元素
def fib(n):
i,a,b=0,0,1
L=[]
while i<n:
L.append(a)
a,b=b,a+b
i+=1
return L
#改用生成器函数
def fib(n)
i,a,b=0,0,1
while i<n:
yield a
a,b=b,a+b
i+=1
#调用函数
n=int(input('请输入个数'))
L=fib(n)
for x in L:
print(x)
当n值很大时,用第一种方法会很占用内存,用生成器则可以节省内存开销。
每次遇到yield语句时,函数会暂停执行,并保存当前所有的运行状态,返回yield后面的值,并在下一次执行next方法时从当前位置继续执行。
def judge(n):
sum=0
c=n
l = []
while c!=0:
i=c%10
l.append(i)
c=c//10
for x in l:
sum+=x**3
yield sum
for i in range(100,1000):
L=judge(i)#L为生成器对象,接受函数的返回值
for x in L:#所有的sum值
if x==i:
print(x)