迭代器与生成器

迭代器

可以直接作用于for循环的数据类型有以下几种：

一类是集合数据类型，如list、tuple、dict、set、str等；

一类是generator，包括生成器和带yield的generator function。

这些可以直接作用于for循环的对象统称为可迭代对象。

迭代器是一个可以记住遍历的位置的对象。

迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

迭代器有两个基本的方法：iter() 和 next()。

字符串，列表或元组对象都可用于创建迭代器

list = [1,2,3,4]
it = iter(list)
print(next(it))
print(next(it))


>>>1
>>>2

迭代器对象可以使用常规for语句进行遍历

list = [1,2,3,4]
it = iter(list)
for i in it:
    print(i,end=" ")


#使用next()

import sys
list = [1,2,3,4]
it = iter(list)

while True:
    try:
        print(next(it))
    except StopIteration:
        sys.exit()
#得到：

1
2
3
4

 

生成器

在 Python 中，使用了 yield 的函数被称为生成器（generator）。

跟普通函数不同的是，生成器是一个返回迭代器的函数，只能用于迭代操作，更简单点理解生成器就是一个迭代器。

在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回yield的值。并在下一次执行 next()方法时从当前位置继续运行。

使用 yield 实现斐波那契数列

import sys
def Fibonacci(n):
    a,b,counter = 0,1,0
    while True:
        if (counter > n):
            return
        yield a
        a,b = b,a+b
        counter += 1
f = Fibonacci(10)

while True:
    try:
        print(next(f),end=" ")
    except StopIteration:
        sys.exit()

#得到

0 1 1 2 3 5 8 13 21 34 55

def triangles(n):
    L = [1]
    while True:
        yield LookupError
        L = [L[x]+L[x+1] for x in range(len(L)-1)]
        L.insert(0,1)
        L.append(1)
        if len(L) > n:
            break
a = triangles(10)

for i in a:
    print(i)


——————————————
#杨辉三角

def angle(n):
    L = [1]
    while True:
        yield L
        L.append(0)
        L = [L[i - 1] + L[i] for i in range(len(L))]
        if len(L) > n:
            break
m = angle(10)
for i in m:
    print(i)