python3: CSV書き込み&読み込み

今日はpythonのCSV読み書きについて学んだ。

ファイルの書き込み

open()で、対象ファイルのファイルオブジェクトを取得し、
DictWriter()の第一引数に指定する。
fieldnamesにはCSVの列名のリストを渡す。
wirtehader()でヘッダ行をCSVに書き込む。
writerow()で、CSVの行を書き込む。

import csv

with open('test.csv', 'w') as csv_file:
    field_names = ['Name', 'HP']
    writer = csv.DictWriter(csv_file, fieldnames=field_names)

    writer.writeheader()

    writer.writerow({'Name': 'Neko', 'HP': 32})
    writer.writerow({'Name': 'Kuma', 'HP': 10})

ファイルの読み込み

DictReader()で、CSVのデータを読み込む事ができる。

with open('test.csv', 'r') as csv_file:
    reader = csv.DictReader(csv_file)
    for row in reader:
        print(row['Name'], row['HP'])