PythonでCSVファイルを読み込む方法

PythonでCSVファイルを読み込む際には、標準ライブラリのcsvを使用します。このライブラリには、CSVファイルの読み取りに使用できるreader()メソッドがあります。

CSVファイルの読み込み

CSVファイルを読み込む基本的なコードは以下の通りです。

import csv

csv_file = open("./TEST_STOCK.csv", "r", encoding="ms932")
f = csv.reader(csv_file, delimiter=",", doublequote=True, lineterminator="\r\n", quotechar='"', skipinitialspace=True)

このコードでは、open関数でCSVファイルを読み込み、そのファイルオブジェクトをcsv.readerに読み込ませています。

エラーハンドリング

実行したときに、AttributeError: module 'csv' has no attribute 'reader'というエラーが発生することがあります。このエラーが発生したときは、csvライブラリが読み込まれていない可能性があります。自身がテスト実行しているpythonファイルのファイル名が「csv.py」になっているとかが原因なので、ファイル名を修正してやれば実行できます。

ファイルのクローズ

openを使用した場合は、必ずcloseしましょう。しなくても、ガベージコレクションによってcloseされますが、closeされるか分からないので、予期しない動きになることがあります。使用するタイミングが終わるとcloseするのが基本的なお作法になります。

withステートメントの使用

openを使用するときは以下のようにwithを使用すると良いです。withを使用すると、そのブロックが終了すると自動的にcloseを実施してくれます。

import csv

with open("./TEST_STOCK.csv", "r", encoding="ms932") as csv_file:
    f = csv.reader(csv_file, delimiter=",", doublequote=True, lineterminator="\r\n", quotechar='"', skipinitialspace=True)