Проверьте, есть ли файлы .format в каталоге

Question

Я некоторое время пытался выяснить, как проверить, есть ли файлы .pkl в данном каталоге.Я проверил веб-сайт и смог найти способы найти файлы в каталоге и перечислить их, но я просто хочу проверить, есть ли они там.

В моем каталоге всего 7 файлов .pkl, как только я создаю один, остальные создаются, чтобы проверить, существуют ли семь из них, будет достаточно проверить, существует ли один.Поэтому я хотел бы проверить, есть ли файл .pkl.

Это работает, если я делаю:

os.path.exists('folder1/folder2/filename.pkl')

Но мне пришлось написать один измои имена файлов.Я хотел бы сделать это без поиска конкретного файла.Я также попытался

os.path.exists('folder1/folder2/*.pkl'),

, но он не работает, так как у меня нет файла с именем * .pkl.

Answer 1

Вы можете использовать:

for dir_path, dir_names, file_names in os.walk(search_dir):
        # Go over all files and folders 
        for file_name in file_names:
            if (file_name.endswith(".pkl")):
                # do something like break after the first one you find

Примечание. Это можно использовать, если вы хотите искать также во всех каталогах с подкаталогами

Если вы хотите искать только один каталог, выможно запустить «для» на os.listdir (путь)

Answer 2

Вы можете использовать модуль python glob (https://docs.python.org/3/library/glob.html)

В частности, glob.glob('folder1/folder2/*.pkl') вернет список всех файлов .pkl в папке 2.