[Python] Read a csv file with a large data size using a generator

import csv


def csv_reader(file, header=False):
    with open(file, "r") as f:
        reader = csv.reader(f)

        if header:
            next(reader)

        yield from reader


def main():
    g = csv_reader("xxx.csv", header=False)
    print(next(g)) # =>First line

if __name__ == '__main__':
    main()

yield from reader

for row in reader:
    yield row

Recommended Posts