python- 迭代器与生成器

Posted 2020-07-22 UnixFBI 运维特工

1.迭代器：

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

特点：

1. 访问者不需要关心迭代器内部的结构，仅需通过next()方法不断去取下一个内容
2. 不能随机访问集合中的某个值 ，只能从头到尾依次访问
3. 访问到一半时不能往回退
4. 便于循环比较大的数据集合，节省内存

a = iter([1,2,3,4,5])
print(a)
输出结果：
<list_iterator object at 0x0000000000B5A5C0>

获取a的值：

方法一：

a = iter([1,2,3,4,5])
for i in a:
    print(i)
输出结果：
1
2
3
4
5

方法二：

a = iter([1,2,3,4,5])
a1 = a.__next__()
print(a1)
a2 = a.__next__()
print(a2)
a3 = a.__next__()
print(a3)
a4 = a.__next__()
print(a4)
a5 = a.__next__()
print(a5)
输出结果：
1
2
3
4
5

 2.生成器：

如果函数中包含yield语法，那这个函数就会变成生成器；

一个函数调用时返回一个迭代器，那这个函数就叫做生成器（generator）；

def func():
    print(‘start‘)
    yield 1
    yield 2
    yield 3
ret = func()
print(ret)
输出结果：
<generator object func at 0x0000000000B4C518>

上面并没有输出start，因为函数里面有yield，就表示 该函数是一个生成器，所以不会执行print(‘start‘)

怎么才能让他执行呢？

方法一：for循环

def func():
    print(‘start‘)
    yield 1
    yield 2
    yield 3
ret = func()
for i in ret:
    print(i)
结果：
start
1
2
3

方法二：使用__next__()迭代

def func():
    print(‘start‘)
    yield 1
    yield 2
    yield 3
ret = func()
a1 = ret.__next__()  #进入函数找到yield，获取yield后面的数据
print(a1)
a2 = ret.__next__() #进入函数找到yield，获取yield后面的数据
print(a2)
a2 = ret.__next__()  #进入函数找到yield，获取yield后面的数据
print(a2)

输出结果：
start
1
2
3