关于生成器的一点笔记
概念
生成器,其本质是在数据需要的时候数据才出现,其他时候待命就好。
如此一样,原本大块的数据,得并非同时需要的数据,是可以被分割成一块块的小数据,这样减去了数据所占用内存空间,也不用消耗计算机提前计算一大块数据的时间。
使用yield语句构建生成器,书上说的使用StopIteration异常来终止生成器,并且不会生成回溯,仅仅是在python2的解释器上。
在python3.7中,StopIteration会生成回溯,并提示错误RuntimeError: generator raised StopIteration而同一份代码在python2则会不报错。
另外,在pyton2中不能使用return来终止生成器,并且还会报错。而在python3中可以使用return来终止生成器,不报错。
建议使用StopIteration来终结生成器,因为其只在python3.7中才会报错,即3+版本还是能正常运行的。如果是使用了循环,那可是用if else和break语句搭配来终止生成器,其可在版本2和版本3中运行。
使用示例
demo1
def generation():
yield 1
yield 2
yield 3
yield 4
if __name__ == '__main__':
a = [i for i in generation()]
print(a) # 运行结果 [1, 2, 3, 4]
demo2
生成器是迭代器的一种