Python

PythonでCSVファイルを扱う

データを保存する時、読み込むときなどにCSVを扱うことがあります。

本稿では、Pythonの標準のモジュールを使って、読み込みや保存の仕方を確認します。公式のリファレンス

読み込み

import csv

with open('data.csv', newline='', encoding='utf-8') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

書き込み

import csv

line = [0, 'TARO', 21]

with open('data.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(line)

書き込み(複数行)

import csv

data = [[0, 'TARO', 21], [1, 'ERI', 20], [2, 'KEN', 20]]

with open('data.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows(data)