python技巧

使用生成器来读取文件(适合大容量存储文件的读取)

2018-08-22  本文已影响2人  陆_志东

使用yield返回数据
注意使用了yield之后函数就变成了生成器,使用的话需要初始化,即加括号

def read_file():
    with open("./oid.txt","r",encoding="utf-8",newline="\n") as f:
        while True:
            data = f.readline().strip()
            if data is None or not data:
                break
            yield data

for data in read_file():
  print(data)
上一篇 下一篇

猜你喜欢

热点阅读