Python11：生成器

生成器

一、列表推导式

列表推导式，可以快速得到需要的列表

缺点：如果一个列表特别的庞大，这样所有的元素直接初始化到内存中，引起大量无用元素占有内存问题

>>> [x for x in range(100)]
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

>>> [i for i in range(100) if i % 2 == 0 ]
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96, 98]

>>> [i*j for i in range(10) for j in range(10)]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 0, 3, 6, 9, 12, 15, 18, 21, 24, 27, 0, 4, 8, 12, 16, 20, 24, 28, 32, 36, 0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 0, 6, 12, 18, 24, 30, 36, 42, 48, 54, 0, 7, 14, 21, 28, 35, 42, 49, 56, 63, 0, 8, 16, 24, 32, 40, 48, 56, 64, 72, 0, 9, 18, 27, 36, 45, 54, 63, 72, 81]

通过列表推导式，我们可以直接创建一个列表。但是，受到内存限制，列表容量肯定是有限的。而且，创建一个包含100万个元素的列表，不仅占用很大的存储空间，如果我们仅仅需要访问前面几个元素，那后面绝大多数元素占用的空间都白白浪费了。所以，如果列表元素可以按照某种算法推算出来，那我们是否可以在循环的过程中不断推算出后续的元素呢？这样就不必创建完整的list，从而节省大量的空间。在Python中，这种一边循环一边计算的机制，称为生成器：generator。

二、列表生成器

[ 列表推导式 ] --> (列表推导式) 就会变成一个列表推导式

使用全局函数next()，每调用一次next()，返回下一个值，直到最后抛出异常

也可以使用循环迭代generator

>>> a = (i for i in range(10))
>>> a
<generator object <genexpr> at 0x0000023202F74200>
>>> next(a)
0
>>> next(a)
1
>>> next(a)
2
>>> next(a)
3
>>>
>>>
>>> for i in a:
...     print(i)
...
4
5
6
7
8
9

生成器对象也存在一个__next__魔方方法，等价于next全局函数

>>> a = (i for i in range(10))
>>> a.__next__
<method-wrapper '__next__' of generator object at 0x0000023202F742E0>
>>> a.__next__()
0
>>> a.__next__()
1

三、函数转换为列表生成器

生成器非常强大。如果推算的算法比较复杂，用类似列表生成式的 for 循环无法实现的时候，还可以用函数来实现。
比如，著名的斐波拉契数列（Fibonacci），除第一个和第二个数外，任意一个数都可由前两个数相加得到：
1, 1, 2, 3, 5, 8, 13, 21, 34, …
斐波拉契数列用列表生成式写不出来，但是，用函数把它打印出来却很容易：

def fibonacci(times):
    num1 = 1
    num2 = 1
    index = 0
    while index < times:
        print(num1, end="")
        print("  ", end="")
        num1, num2 = num2, num1+num2
        index += 1


fibonacci(5)

"""
结果：
1  1  2  3  5  
"""

如果函数中使用yield关键字，那么这个函数的返回值就是一个生成器

def fibonacci(times):
    num1 = 1
    num2 = 1
    index = 0
    while index < times:
        yield num1
        num1, num2 = num2, num1+num2
        index += 1


result = fibonacci(5)
print(result)
print(next(result))
for i in result:
    print(i)

"""
结果：
<generator object fibonacci at 0x000001AFD261DC10>
1	#next(result)的输出
1	#循环的输出
2
3
5
"""