十python生成器和迭代器

Posted 2020-11-02 caibaofei

一、列表生成式：

1、生成一个列表：

list = [ i*2 for i in range(10)]       #使用列表生成式生成一个列表，占据内存空间，当数量大时会造成很大的浪费。

print(list)

 

2、使用生成器生成列表：（generator）

list1 = ( i*2 for i in range(10))      #将生成列表的公式存在变量中，需要的时候取一下，（注：缺点是只能往后一个个取）

print(list1.__next__())                    #使用__next__取数据，一次取一个数，当取完时抛出StopIteration错误

for i in list1:              #由于一次一次的取比较麻烦，我们还可以使用for循环来取

    print(i)                

 

3、如果需要推算的结果使用列表无法生成式，我们还可以使用函数来制作生成器，如斐波拉契数列。

斐波拉契数列：（fibonacci）

介绍：斐波拉契数列，是一种除了第一个和第二个数外，任意一个数都可由前两个数相加得到

由python函数实现斐波拉契数列：

def fib(a,b,max):                           #a是指斐波拉契中的第一个数，b指斐波拉契中的第二个数，max指相加几次

    n=0

    while n<max:

        print(b)

        a,b=b,a+b

        n=n+1

        return‘done‘

fib(10,15,12)

如果要将上面的函数变成生成器，只要将print(b)改成yield b就可以了

yield：每次调用到yield时返回结果，并保存现在的位置，再次执行时从保存的位置继续。

def fib(a,b,max):

    n = 0

    while n < max:

        yield b

        a, b = b, a + b

        n = n + 1

    return ‘done‘

 

f = (fib(10,15,12))                #将函数指定为一个变量，通过变量来取值，不然无法边取值边传参。

print(f.__next__())

print(f.__next__())

print(f.__next__())