Python and CSV

Python has a very nice module called csv which makes working with CSV very easy. This mini article is only a reminder for me so that I can easily find how to use it when I forget once again how it is used exactly.

Reading CSV files

try:
    from future.builtins import open
except:
    pass

import csv

with open("eggs.csv", "rt", newline="") as csvfile:
    csvreader = csv.reader(csvfile, delimiter=";", quotechar='"')
    next(csvreader, None)  # skip the headers
    for row in csvreader:
        print(", ".join(row))

Writing CSV files

try:
    from future.builtins import open
except:
    pass

import csv

with open("eggs.csv", "w", newline="") as csvfile:
    csvwriter = csv.writer(
        csvfile, delimiter=";", quotechar='"', quoting=csv.QUOTE_MINIMAL
    )
    csvwriter.writerow(["Spam"] * 5 + ["Baked Beans"])  # Write header
    for row in container:
        csvwriter.writerow(row)  # Write data

See also

Published


by

Category

Code

Tags

Contact