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

Question

我有一个文件夹，其中包含不同类型的文件。我只需要处理图像文件(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

您不能像glob.glob('*.{JPG, png..})这样进行复杂的通配符全局化，因为如果您查看一下您将看到的源代码，

def glob(pathname):
    """Return a list of paths matching a pathname pattern.
    ....
    """
    return list(iglob(pathname))

如果你找到iglob的源头，你就会看到，

def iglob(pathname):
    ....
    ....
    dirname, basename = os.path.split(pathname)

    # voila, here, our complex glob wildcard will certainly break, and can't be used :)

因此，您只能使用glob :)执行简单的glob必应操作。

Answer 2

使用路径库

from pathlib import Path

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