Python By Hank

Python 腳本效能:Generator 的使用

節省記憶體

處理大數據或大型 Log 檔時,使用 yield 建立 Generator,而不是用 List Comprehension 一次把資料載入記憶體。

# Good
def read_large_file(file_object):
    while True:
        data = file_object.readline()
        if not data:
            break
        yield data