Как вернуть файлы только с определенным типом файлов?

Это мой код:

import os

def read_file(path):
    name_file = os.listdir(path)
    return name_file

path = "/home/ich/Schreibtisch/fotki"
print(read_file(path)) 

Это возвращает только все файлы в папке в списке. Но я хочу, чтобы они закончились на ".jpg". В папке много файлов (.jpg, .txt, ...), но мне нужны только файлы .jpg.

Stackoverflow - англоязычный сайт. Пожалуйста, задавайте свои вопросы только на английском языке.

zvone 26.09.2018 20:57

Их список можно получить с помощью модуля glob. то есть jpg_files = glob.glob(os.path.join(path, "*.jpg")). Обратите внимание, что результатом будет пустой список, если нет файлов, соответствующих шаблону.

martineau 27.09.2018 02:10
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
2
2
82
1

Ответы 1

Сделайте это с os.listdir и .endswith

import os

path = "/home/ich/Schreibtisch/fotki"

for file in os.listdir(path):
    if file.endswith(".jpg"):
        print(os.path.join(path, file))

Я могу только предполагать, но, возможно, проигравшему не понравилось легкое нарушение принципа DRY (вы повторяете '/mydir').

timgeb 26.09.2018 21:04

Спасибо за подсказку. Кстати, есть ли книга или хорошее место, чтобы получить обзор таких принципов?

mrk 26.09.2018 21:06

Я считаю, что неравнодушные программисты со временем поймут эти вещи автоматически. Однако, если вам необходимо прочитать книгу, есть классика, такая как Code Complete.

timgeb 26.09.2018 21:20

Другие вопросы по теме