выбрать случайные файлы PNG или JPEG - python - PullRequest
0 голосов
/ 17 марта 2020

Я пытаюсь написать программу, которая выбирает случайную фотографию, и я создал этот код:

random_image_path = random.choice([x for x in os.listdir(path) if os.path.isfile(os.path.join(path, x))])
print(random_image_path)

, но после запуска программы она иногда печатает файлы ".ini" ... что делать, если мне нужны только файлы .png или .jpeg? спасибо!

1 Ответ

2 голосов
/ 17 марта 2020

Добавить фильтр, включающий только файлы с такими расширениями:

x for x in ... if x[-4:].lower() in ('.jpg', '.png')
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...