【Python】CSVファイルを読み書きする方法

2019年7月3日
2020年4月11日
Python

Python

こんにちは！

皆さんはCSVファイルからデータを読み込んだり、 CSVファイルへデータを書き込みたいと思ったことはありますか？

CSVファイルはエクセルなどでも開くことができたり、よく使われているデータ形式なので、プログラミングを勉強している人なら触れたことがあるデータ形式ではないかと思います。

今回はPythonでCSVファイルを読み書きする方法として、以下の2つの方法を紹介します。

with文を用いる方法
with文を用いない方法

1 CSVファイルの読み込み
- 1.1 with文を用いる方法
- 1.2 with文を用いない方法
2 CSVファイルへの書き込み
- 2.1 with文を用いる方法
- 2.2 with文を用いない方法
3 まとめ

CSVファイルの読み込み

CSVファイルの読み込みには、「csvモジュール」を使用します。

csvモジュールはPythonの標準ライブラリなので、インストールの必要はありません。

今回はCSVファイルの読み込み・書き込み方法を、with文を用いる方法、with文を用いない方法の２通り紹介します。

with文については、以下の記事を参考にしましょう。

参考記事

【Python】with文の使い方

こんにちは！皆さんは「with文」というものをご存知ですか？ with文は、with文を用いなくても他の書き方で書けるため、知っていても使い方は理解していない、といった方もいるかと思います。 with文はファイルのオー[…]

with文を用いる方法

以下のプログラムは、with文を用いた、CSVファイルを読み込むプログラムです。

import csv

data1 = 'file1.csv'

with open(data1, 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # ヘッダーを読み飛ばす場合

    for row in reader:
        print(row)

実行結果(file1.csvに下記のデータが入っている場合)

['160', '60']
['170', '70']

with文を用いない方法

次のプログラムは、with文を用いない、CSVファイルを読み込むプログラムをです。

import csv

data1 = 'file1.csv'

f = open(data1, "r")
reader = csv.reader(f)
header = next(reader) # ヘッダーを読み飛ばす場合

for row in reader:
　　print(row)
f.close()

実行結果(file1.csvに下記のデータが入っている場合)

['160', '60']
['170', '70']

with文を用いる場合と用いない場合の違いに注目してみましょう。

1.1のwith文を用いる場合との大きな違いがわかりますか？

そうです！最後に「f.close()」によってファイルを閉じる必要があります。

忘れないように注意してください。忘れん坊の方はwith文を用いるようにした方が良いかもしれませんねw

CSVファイルへの書き込み

CSVファイルへの書き込みも、「csvモジュール」を使用します。

with文を用いる方法

次のプログラムは、with文を用いた、CSVファイルへの書き込みを行うプログラムです。

import csv

data2 = 'file2.csv'

header = ['身長', '体重']
data = [[160, 60], [170, 70]] # 2次元配列

with open(data2, 'w') as f:
    writer = csv.writer(f, lineterminator='\n')
    writer.writerow(header)
    writer.writerows(data)

with文を用いない方法

次のプログラムは、with文を用いない、CSVファイルへの書き込みを行うプログラムです。

import csv

data2 = 'file2.csv'

header = ['身長', '体重']
data = [[160, 60], [170, 70]] # 2次元配列

f = open(data2, 'w')

writer = csv.writer(f, lineterminator='\n')
writer.writerow(header)
writer.writerows(data)

f.close()

作成されたfile2.csv（例）