Pythonで特定のディレクトリ内のファイルを反復処理する方法をわかりやすく解説

os.listdir()でディレクトリの中身を一括取得する

Pythonで特定のディレクトリ内のファイルやフォルダを反復処理したい場合、最も基本的な方法が標準ライブラリのos.listdir()を使うことです。os.listdir(my_path)を実行すると、指定したパス（my_path）に含まれるファイルとディレクトリの一覧がリスト形式で返されます。

以下は、カレントディレクトリ（'.'）の中身を取得する例です。

>>> import os
>>> os.listdir('.')
['DLLs', 'Doc', 'etc', 'include', 'Lib', 'libs', 'LICENSE.txt', 'NEWS.txt', 'python.exe', 'pythonw.exe', 'README.txt', 'Scripts', 'share', 'tcl', 'Tools', 'w9xpopen.exe']

このように、ファイルだけでなくサブディレクトリもすべて含まれる点に注意してください。

os.path.isfile()でファイルだけを抽出する

ディレクトリではなく「ファイルだけ」を取得したい場合は、リスト内包表記とos.path.isfile()を組み合わせてフィルタリングします。パスを正しく結合するためにos.path.join()を使うのがポイントです。

>>> import os
>>> file_list = [f for f in os.listdir('.') if os.path.isfile(os.path.join('.', f))]
>>> print(file_list)
['LICENSE.txt', 'NEWS.txt', 'python.exe', 'pythonw.exe', 'README.txt', 'w9xpopen.exe']

このコードでは、各要素に対してos.path.isfile()で判定を行い、ファイルであるものだけを新しいリストに格納しています。逆にディレクトリだけを取得したい場合は、os.path.isdir()に置き換えるだけで対応できます。

よりモダンな代替手段：pathlibとos.scandir()

Python 3.4以降では、オブジェクト指向的な操作が可能なpathlibモジュールも便利です。

from pathlib import Path

p = Path('.')
files = [f for f in p.iterdir() if f.is_file()]
print(files)

また、大量のファイルを扱う場合は、os.scandir()を使うとos.listdir()よりも効率的に処理できます。用途に応じてこれらの方法を使い分けるとよいでしょう。