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では正規表現が使えないため、少し遠回りする形になった。