从文件夹中选择所有文件 - 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文件。