Pythonのglobモジュールで複数の拡張子を指定してファイル一覧を取得する
カレントディレクトリの「画像」ディレクトリの中のファイルの中から
拡張子が「jpg」と「png」のファイルの一覧を取得する。
方法1
import glob types = ('jpg', 'png') files = [] for t in types: files += glob.glob('./画像/*.' + t) for file in files: print(file)
方法2
import glob for file in glob.glob('./画像/*'): type = file[-3:] if type in ('jpg', 'png'): print(file)
方法3
import glob import re for file in glob.glob('./画像/*'): if re.match('.+\.(png|jpg)$', file): print(file)