pythonで特定のディレクトリから画像ファイルのリストを取得(拡張子で判別)
pythonで特定のディレクトリから画像ファイルのリストを取得する。
画像ファイルかどうかは拡張子で判定、今回は.jpg, .png, .bmpを画像ファイルとする(大文字小文字は区別しない)
# 正規表現でディレクトリから画像ファイルのリストを取得 import os import re path = "data/" pattern = ".*\.(jpg|png|bmp)" files = [f for f in os.listdir(path) if re.search(pattern, f, re.IGNORECASE)] # 大小文字無視 print(files)
結果
['001.png', '002.png', '003.png']
globでは正規表現が使えないため、少し遠回りする形になった。