[Python] Lire un fichier csv avec une grande taille de données à l'aide d'un générateur

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)) # =>Première ligne

if __name__ == '__main__':
    main()

yield from reader

for row in reader:
    yield row

Recommended Posts