Python3入門 – ファイル入出力

write()によるテキストファイルへの書き込み

ファイルへ書き込むテキストとして次の文字列を使います。

>>> text = '''海が光る
... 半分から
... こっちが光る
... '''

通常、ファイルへの書き込みにはモード「wt」を使います。

>>> fout = open('sample.txt', 'wt')
>>> fout.write(text)
17
>>> fout.close()

open()関数はファイルのオープンに成功するとファイルオブジェクトを返します。

テキストファイルへの書き込みは、このはファイルオブジェクトのwrite()関数で行います。write()関数は書き込んだ文字数を返します。

すでに存在するファイルを「wt」で開くと元の内容が失われます。これが望んだことでなければ、開く前にos.path.exists()関数でファイルが存在するかどうかを確認します。

print()によるテキストファルへの書き込み

print()でテキストファイルに書き込むこともできます。

print()はデフォルトで標準出力（sys.stdout）へテキストを書き出しますが、引数fileへ任意のファイルオブジェクトを渡して、そこへ書き込むこともできます。

>>> fout = open('sample.txt', 'wt')
>>> print(text, file=fout, end='')
>>> fout.close()

ここでは最後に改行文字（'\n'）を追加して欲しくないので、「end=''」を指定しています。

print()はwrite()より高水準な出力関数です。複数のオブジェクトが渡されたときにセパレータ（引数sep）を間に挟んだり、最後に文字列(引数end）を追加することができます。

>>> print('one', 'two', 'three', sep=' ', end='\n')
one two three

この例では説明のために明示的に「sep=' '」や「end='\n'」を渡していますが、これらはデフォルトなので省略しても結果は変わりません。これらの引数にお好みの文字列を指定して、デフォルトの動作を変更できます。

セパレータや末尾に文字列を追加したければwrite()の代わりにprint()を使うといいでしょう。

ファイルの上書きを禁止する

間違ってファイルを上書きしたくないならモード「x」を使ってファイルを開きます。

>>> fout = open('sample.txt', 'xt')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
FileExistsError: [Errno 17] File exists: 'sample.txt'

ファイルが既に存在していればFileExistsErrorが発生し、ファイルのオープンに失敗します。

既存のファイルを上書きしないことを除けば、モード「w」で開いたときと使い方は同じです。

既存のテキストファイルへ追記する

既存のファイルにテキストを追記したいならモード「at」でファイルを開きます。

>>> fout = open('sample.txt', 'at')
>>> print('これは追記テキスト', file=fout)
>>> fout.close()

これは既存のテキストファイルの最後からテキストを書き込み始めます。

テキストファイルを一度に読み込む

最初はread()関数を使ってテキストファイルを読んでみましょう。read()関数を使うとファイルの内容をすべて一度に読み込みます。

>>> fin = open('sample.txt', 'rt')
>>> content = fin.read()
>>> fin.close()
>>> content
'海が光る\n半分から\nこっちが光る\n'

read()関数の引数を省略するとファイルの内容を一度にすべて読み込みます。これはサイズが大きいファイルを読み込む場合、大量のメモリを消費します。

一度に読み込む文字数を制限したければ、read()関数の引数に一度に読みだす文字数を渡すことができます。

>>> chunks = []
>>> fin = open('sample.txt', 'rt')
>>> while True:
...   chunk = fin.read(5)
...   if not chunk:
...     break
...   chunks.append(chunk)
...
>>> fin.close()
>>> chunks
['海が光る\n', '半分から\n', 'こっちが光', 'る\n']

この例では一度に読み出すのは5文字です。これを繰り返し実施しています。

ファイルの内容をすべて読んだ後にread()関数を呼び出すと空文字列が返されます。

空文字列をifの条件で使うとFalseと評価されます。この例ではchunkが空文字列のとき、if文本体のbreakが実行されwhileのループを抜けます。

テキストファイルを1行ずつ読み込む

readline()関数を使うと、ファイルを1行ずつ読みだすことができます。readline()関数は呼び出されるごとに次の1行を返します。

>>> lines = []
>>> fin = open('sample.txt', 'rt')
>>> while True:
...   line = fin.readline()
...   if not line:
...     break
...   lines.append(line)
...
>>> fin.close()
>>> lines
['海が光る\n', '半分から\n', 'こっちが光る\n']

ファイルの最後まで読んだ後にreadline()関数を呼び出すと、read()関数と同様に空文字列が返されます。

返された行の末尾の改行コード（'\n'）は残されます。ただし、ファイルの最後が改行コードで終わっていなければ、最後の行にはもちろん改行コードはありません。

for文でファイルオブジェクトを使う

ファイルオブジェクトはイテラブルなのでfor文で使うことができます。

for文で使うとreadline()と同様、繰り返すごとに次の1行が返されます。

>>> data = []
>>> fin = open('sample.txt', 'rt')
>>> for line in fin:
...   data.append(line.rstrip('\n'))
...
>>> fin.close()
>>> data
['海が光る', '半分から', 'こっちが光る']

最後の行まで読み込むとfor文を抜けます。

readline()関数と同様、返された行の末尾の改行コード（'\n'）は残されます。

この例では行末の改行コードをrstrip()を使って削除しています。

すべての行をリストに読み込む

readline()関数の例では1行ずつ読み出してリストに追加していました。本当にこのようなデータが欲しければreadlines()を使った方が簡単です。

>>> fin = open('sample.txt', 'rt')
>>> lines2 = fin.readlines()
>>> fin.close()
>>> lines2
['海が光る\n', '半分から\n', 'こっちが光る\n']

readlines()関数は1行ずつ読み込み、読み込んだ行をリストとして返します。

with文による自動的なファイルのクローズ

ファイルを開いたら必ず閉じる必要があることは前述しましたが、Pythonではこれを確実に行うための機構が用意されています。

それにはPythonのwith文を使います。with文を使うと最初の書き込みの例は次のように書くことができます。

>>> with open('relativity', 'wt') as fout:
...     fout.write(poem)
...

open()関数をwith文で使うと、コンテキストマネージャが呼び出され、with文のブロックはそのランタイムコンテキストで実行されます。

と、このような難しい話はとりあえず置いておいて次の要点を押さえておきましょう。

1. with文のopen()関数の呼び出しが成功すると、open()関数が返すファイルオブジェクトはasの後の変数（変数名は任意）へ代入される。
2. with文本体のブロックでは、これまで通りファイルオブジェクトを使ってファイルを読み書きする。
3. with文のブロックを抜けるとき、コンテキストマネージャの終了処理が実行される。これはwithブロックを正常に抜けるか、または例外によって抜けるかに関わらずファイルを必ず閉じます。

このようwith文を使うと、もはや明示的にclose()関数を呼び出す必要はなくなります。

引数newlineと改行コード

ここまで行末の改行コードについては触れて来ませんでした。ご存知かもしれませんが、UnixやMacではデフォルトの改行コードには'\n'が使われ、Windowsでは'\r\n'が使われています。

PythonでWindowsのテキストファイルを読み込んだとき、行の最後には'\r\n'という文字列が付くのではないか？と気になったかもしれません。

しかし、これまでの例ではテキストファイルの改行コードとして'\n'と'\r\n'のどちらが使われていいたとしても、読み込んだ行の末尾は必ず'\n'で終わります（ただし、最後の行は前述した通り'\n'が付かないことがあります）。

これはopen()関数の引数newlineが省略されているときのデフォルトの動作です。newline引数を省略すると、読み込んテキストの改行コード（'\n'や'\r\n'）は1文字の'\n'へ変換されます。

テキストファイルへ書き込むときにopen()関数の引数newlineを省略すると、反対の変換が行われます。書き込む文字列の'\n'は、システムデフォルトの改行コード（UnixやMacなら'\n'、Windowsなら'\r\n'）へ変換されてからファイルへ書き込まれます。

このようなopen()関数のデフォルトの動作を使うと、実際のファイルの改行コードがなんであるか意識する必要はなく、プログラムでは常に改行コードが'\n'であるとして処理ができるという利点があります。

このデフォルトの動作を変更したい場合は、引数newlineに別の値を指定します。

これについては「Python3でファイルを読み書きするとき、改行コードはどうなるの？」に詳しく書きましたので参照してください。

引数encodingと文字コード（文字エンコーディング）

ファイルを開くときにファイルのエンコードやデコードするための文字エンコーディング（文字コードや文字符号化方式ともいう）を引数encodingに指定できます。よく使われる文字エンコーディングにはUTF-8やcp932などがあります。

引数encodingを省略するとOSデフォルトのエンコーディング（正確にはlocale.getpreferredencoding()が返すもの）が使われます。

特定の文字エンコーディングでファイルを読み書きするときは、引数encodingに指定します。指定できる文字エンコーディングはcodecsモジュールのドキュメントを参照してください。

なお、一般にWindowsでShift_JISを使うのは厳密には正しくありません。「Shift_JISとCP932とWindows-31Jの違いを整理した」の記事を読んでいただければ、この言葉の意味やcp932について理解いただけるでしょう。