1. ホーム 
  2. 備忘録 
  3. Python

CSVファイル

CSVファイルの読み書き

PythonでCSVファイルの読み書きを行うには、csvモジュールをインポートする

csvモジュールの使い方を以下に記載する

lesson.py

import csv

# Windowsの場合、改行が \r\n となりcsv読み込みの際に2行改行されてしまう
# newline=''とすることで問題を回避できる
with open('test.csv', 'w', newline='') as csv_file:
    fieldnames = ['Name', 'Count']
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'Name': 'A', 'Count': 1})
    writer.writerow({'Name': 'B', 'Count': 2})

with open('test.csv', 'r') as csv_file:
    reader = csv.DictReader(csv_file)
    for row in reader:
        print(row['Name'], row['Count'])
        # 結果
        # Name,Count
        # A,1
        # B,2