我的Python成长之路---第四天---Python基础(14)---2016年1月23日(寒风刺骨)

时间:2022-01-02 21:33:38

一、生成器和迭代器

1、迭代器

迭代器是访问集合元素的一种方式。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退,不过这也没什么,因为人们很少在迭代途中往后退。另外,迭代器的一大优点是不要求事先准备好整个迭代过程中所有的元素。迭代器仅仅在迭代到某个元素时才计算该元素,而在这之前或之后,元素可以不存在或者被销毁。这个特点使得它特别适合用于遍历一些巨大的或是无限的集合,比如几个G的文件。

特点:

  1. 访问者不需要关心迭代器内部的结构,仅需通过__next()__(Python2.x 为next())方法不断去取下一个内容
  2. 不能随机访问集合中的某个值 ,只能从头到尾依次访问,也就是不能像列表一样通过下标等取到任意位置的值,一般多用用于for in 循环进行遍历
  3. 访问到一半时不能往回退
  4. 便于循环比较大的数据集合,节省内存
    Python内置函数iter就是一个简单的帮助我们创造一个迭代器的内置函数。
>>> it = iter([1, 2, 3, 4]) # 通过内置函数iter及列表生成了一个迭代器
>>> it
<list_iterator object at 0x7faa075f4ef0>
>>> it.__next__() # 通过__next方法取得一个值
1 >>> it.__next__()
4
>>> it.__next__() # 迭代器一个只有4个元素,如果此时再继续使用next方法,就会报错
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
    其实事实上,我们很少会使用__next__()方法一个一个的去取值,多数情况下都是使用for in循环进行遍历
>>> it = iter([1, 2, 3, 4])
>>> for i in it:
... print(i)
...
1
2
3
4

2、生成器

    生成器就是一个函数调用时返回一个迭代器,这个函数就叫做生成器。说白了就是生成器生成迭代器。与普通函数的区别就在于yield关键字
    我们用斐波拉契数列来说明生成器和普通函数的区别
    普通函数实现
def fib(max):
n, a, b = 0, 0, 1
res = []
while n < max:
res.append(b)
a, b = b, a + b
n = n + 1
return res
res = fib(6)
print(res)
for i in res:
print(i)
执行结果
[1, 1, 2, 3, 5, 8]
1
1
2
3
5
8
    生成器
def fib2(max):
n, a, b = 0, 0, 1
while n < max:
yield b
a, b = b, a + b
n = n + 1
print(res)
for i in res:
print(i)
执行结果
<generator object fib2 at 0x000001EF306570F8> # 可以看出返回的是一个生成器对象,并且生成器内部的东西没有马上执行,而是执行到yield的时候就返回一个迭代器
1
1
2
3
5
8
    说明:虽然从结果上看上去是一样的唯一区别就是一个返回的是一个列表,一个是生成器,但实际上函数调用的时候是一次执行函数体的内容,并将结果一次性返回,而生成器不是,生成器只有执行__next__方法或for in循环才会现执行,遇到yield的时候返回当前计算结果,等待下次的调用。这样的好处当要返回的列表特别大的时候,节省内存地址,并且使用函数相当于执行了两次循环(第一次生成列表,第二次遍历列表)而生成迭代器的方式一边执行遍历一边计算结果,效率更高,只相当于执行了一次循环。
 
    这个yield的主要效果呢,就是可以使函数中断,并保存中断状态,中断后,代码可以继续往下执行,过一段时间还可以再重新调用这个函数,从上次yield的下一句开始执行。
 
    另外yield不仅可以返回值,同样可以接收值,我们可以用这个特性实现一个在单线程 的情况下实现并发的效果
 import time
def consumer(name):
print("%s 准备吃包子啦!" %name)
while True:
baozi = yield print("包子[%s]来了,被[%s]吃了!" %(baozi,name)) def producer(name):
c = consumer('A')
c2 = consumer('B')
c.__next__()
c2.__next__()
print("老子开始准备做包子啦!")
for i in range(10):
time.sleep(1)
print("做了2个包子!")
c.send(i)
c2.send(i) producer("alex")

  执行结果

A 准备吃包子
B 准备吃包子
老子开始准备吃包子了
做了2个包子
包子[0]来了,被[A]吃了!
包子[0]来了,被[B]吃了!
做了2个包子
包子[1]来了,被[A]吃了!
包子[1]来了,被[B]吃了!
...