Шукаємо файл у папці за допомогою Python

Інколи за допомогою Python потрібно опрацьовувати файли у папці, але не всі. Потрібні лише деякого типу(розширення). Для цього нам на допомогу приходить стандартний модуль  os
import os
У нього є функція listdir(), що поверне нам список файлів у поточній папці у вигляді списку з розширенням
dir_list = os.listdir()
Хоча ми можемо додати у функцію listdir і якийсь шлях до папки:
path = "C://Users//Vanshi//Desktop//gfg"
dir_list = os.listdir(path)

Та зазвичай нам не потрібні всі файли для опрацювання. Наприклад, нам не потрібні файли, що мають розширення 'py', тоді нам потрібно опрацювати весь отриманий список. Пропустити його через цикл. І перевірити чи відповідає розширення нашого файлу вимогам.

Створюємо пустий список
pic = [] 
За допомогою циклу for проходимо по всіх елементах отриманого списку файлів
for part in dir_list:
І якщо відтинок (останні 2 символи) нашого елемента дорівнюють умові (не дорівнює 'py')
if part[-2::] != 'py':
Тоді ми цей елемента додаємо у наш пустий новостворений список
pic.append(part)
Також можна провести тест, і додати відтяту частинку у наш список. Це зручно для аналізу роботи коду. І при потребі можна це за коментувати, або видалити
pic.append(part[-2::]) 

Весь код буде виглядати так:

import os
dir_list = os.listdir()
pic = [] 
for part in dir_list:
 if part[-2::] != 'py':
  pic.append(part)
  #pic.append(part[-2::]) 

Надіюсь матеріал стане в нагоді ;-)

Коментарі