python - 使用一个 glob.glob 而不是多个 glob.glob

Question

这个问题已经有答案了:

奥 git _a (40 个回答)

已关闭 4 年前。

我有一个包含不同类型文件的文件夹。我只需要处理图像文件(jpg、png、jpeg、JPG...)。为此，我使用 glob.glob 只获取带有“jpg”、“png”等扩展名的文件。
问题是我必须多次使用 glob.glob (与图像扩展名的数量一样多)才能获取所有图像:

import glob
images = glob.glob('tests/*.jpg') + glob.glob('tests/*.jpeg') + glob.glob('tests/*.png') + glob.glob('tests/*.JPG')  

有没有办法以更优化的形式使用 glob.glob (例如 glob.glob('path/*.{'jpg', 'png', 'jpeg', 'JPG'} )或者是否有更简单的函数来使用它而不是 glob ？

Answer 1

使用pathlib :

from pathlib import Path

extensions = ['.jpg', '.png', '.jpeg']
images = [x for x in Path('tests').iterdir() if x.suffix.lower() in extensions]