python生成器和yield关键字

体验

下列代码用于先体验普通列表推导式和生成器的差别:

import time

def get_time(func):
    def wraper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print("Spend:", end_time - start_time)
        return result
    return wraper

@get_time
def _list(n):
    l = [i*i*i for i in range(n)]

@get_time
def _generator(n):
    ge = (i*i*i for i in range(n))

@get_time
def _list_print(l1):
    for i in l1:
        print(end='')

@get_time
def _ge_print(ge):
    for i in ge:
        print(end='')

n = 100000
print('list 生成耗时：')
_list(n)
print('生成器 生成耗时：')
_generator(n)

l1 = [i*i*i for i in range(n)]
ge = (i*i*i for i in range(n))
# print(l1)
# print(ge)
print('list遍历耗时：')
_list_print(l1)
print('生成器遍历耗时：')
_ge_print(ge)

结论：
生成速度：生成器>列表
for_in_循环遍历：1、速度方面：列表>生成器；2、内存占用方面：列表>生成器
总的来说，生成器就是用于降低内存消耗的。

自定义生成器（yield关键字）

参考https://blog.csdn.net/mieleizhi0522/article/details/82142856/