从文件夹中选择所有文件 - Python

Question

我需要从文件夹test_image 中读取所有图像，为此，所有图像的名称必须具有相同的前缀。

PATH_TO_TEST_IMAGES_DIR = 'test_images'
TEST_IMAGE_PATHS = [ os.path.join(PATH_TO_TEST_IMAGES_DIR, 'image{}.jpg'.format(i)) for i in range(1, 21) ]

我想选择另一个文件夹。在不将所有名称更改为该格式的情况下，如何做到这一点？

Answer 1

import glob, os

PATH_TO_TEST_IMAGES_DIR = 'dir'
searchstr = os.path.join(PATH_TO_TEST_IMAGES_DIR, '*.jpg')
list_of_images = glob.glob(searchstr)

使用 glob，假设您只是在查找所有 .jpg 文件。