Python3 Iterator and Generator

Python3 Iterator and Generator

iterator

主要是利用 iter 函数

>>> list=[1,2,3,4]
>>> it = iter(list)    # 创建迭代器对象
>>> print (next(it))   # 输出迭代器的下一个元素
1
>>> print (next(it))
2
>>>

generator

# coding:utf8

def load_file(fname):
    with open(fname, encoding='utf8') as f:
        for line in f:
            yield line.strip()

if __name__ == '__main__':
    zh = load_file('zh.txt')
    print(next(zh))

使用上述程序，读取如下文件（zh.txt）

一对丹顶鹤正监视着它们的筑巢领地
一对乌鸦飞到我们屋顶上的巢里，它们好像专门为拉木而来的。
一对乖乖仔开着老爸的车子。
......

程序输出：

一对丹顶鹤正监视着它们的筑巢领地

Reference

Python3 迭代器与生成器