Pythonでcsvを扱う方法

はじめに

機械学習関連の開発で、大量に出力されたCSVデータを扱う機会があったので、Pythonでcsvデータを扱う方法を備忘録としてまとめてみました。

CSVファイルの中身を1行ずつ表示する

csvファイルを扱うにはpythonに標準的に備わっているcsvモジュールを使用します。
csvファイルを指定して、行の内容をリストとして表示するには以下のように書きます。

import csv

#読み出したいファイルのパス
f = open("C:\\Users\\redpe\\Desktop\\XXXXX.csv")
reader = csv.reader(f)

#1行ずつ出力させる
for row in reader:
     print(row)

f.close()

ヘッダーの行を読み飛ばす

csvファイルを処理する際に、先頭のヘッダ部分は使わないことがあります。
その時は以下のように読み飛ばす処理を入れます。

import csv

#読み出したいファイルのパス
f = open("C:\\Users\\redpe\\Desktop\\XXXXX.csv")
reader = csv.reader(f)

#1行ずつ出力させる
for row in reader:

     #1行目を読み飛ばす場合
     if reader.line_num == 1:
         continue

     print(row)

f.close()

CSVファイルにデータを追記する

もともとのCSVファイルにデータを1行追加したい場合には以下のようにします。
指定したCSVファイルに ['2021/6/13', '11:00', '30', '60', 'piglet']の行を追加しています。

import csv

#追記したファイルのパス
f = open("C:\\Users\\redpe\\Desktop\\XXXXX.csv",mode="a",newline="")
writer = csv.writer(f)

#追記したいデータリストを作成
data = ['2021/6/13', '11:00', '30', '60', 'piglet']

#データを追記する
writer.writerow(data)

f.close()