Pythonでディレクトリー内から特定の拡張子のファイルを取得
listdirを使ってディレクトリー内のファイルを取得し、リスト内包表記を使って条件付きでファイルを読み込む。 if文を変えれば拡張子に限定されない。 例えば、接頭語を指定するとか。
import os def get_target_files(dir_path='./image_dir/'): files = [f for f in os.listdir(dir_path) if f[-4:]=='.png'] return files
コードは一応、ここに置いておきます。
>>> import test >>> test.get_target_files() ['img_000000.png', 'img_000001.png', 'img_000002.png', ...]