Python:csvファイルを読み込み・書き込みする方法

lines Python Tips

csvファイルを読み込む方法

with句を使って以下のように読み込むことができます。

import csv

with open('test.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # ヘッダーがある場合
    for row in reader:
        print(row)

csvファイルを書き込む方法

writerowで1行ずつリストを書き込めます。

import csv

with open('test.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(['a', 'b', 'c'])
    writer.writerow([0, 1, 2])

writerowsで二次元配列を一気に書き込めます。

import csv

with open('test.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerows([['a', 'b', 'c'], [0, 1, 2]])