主页 / 问题 / 765338
Accepted
pynista
Asked:2020-12-31 16:54:54 +0000 UTC2020-12-31 16:54:54 +0000 UTC

如何让 glob.glob() 查找不同格式的图像？

  • 772

该目录包含散布着其他文件的图像文件。图像文件名是数字（例如234.jpg, 3456.jpeg, 111111.png, 4.bmp）。

jpg我收到 带有如下扩展名的文件：

image_files = glob.glob(''.join([dir, '/', '[0-9]*.jpg']))

如何扩展此代码以获取带有扩展名的文件png, jpeg, bmp, jpg？

python

1 个回答

  1. Best Answer

    可以使用str.endswith()（未​​测试）过滤掉所需的文件：

    import os

for entry in os.scandir(dir):
    if (entry.stem.isdecimal()
        and entry.name.endswith(('.png', '.jpeg', '.bmp', '.jpg'))):
        print(entry.path)

    或者，您可以使用正则表达式：

    import os
import re

found_image = re.compile(r'\d+\.(?:jpeg|jpg|bmp|png)').fullmatch
for entry in os.scandir(dir):
   if found_image(entry.name):
       print(entry.path)
    • 1

相关问题