Я намагаюся отримати список файлів у каталозі за допомогою Python, але я не хочу список ВСІХ файлів.

Я, по суті, хочу, це можливість робити щось на кшталт наступного, але використовуючи Python, а не виконувати ls.

ls 145592*.jpg

Якщо для цього немає вбудованого методу, я зараз думаю створити цикл для повторення результатів os.listdir()і додати всі відповідні файли до нового списку.

Однак у цьому каталозі багато файлів, і тому я сподіваюся, що існує більш ефективний метод (або вбудований метод).

glob.glob('145592*.jpg')

glob.glob()це, безумовно, спосіб зробити це (за Ігнасіо). Однак якщо вам потрібне більш складне узгодження, ви можете зробити це з розумінням списку і re.match(), як-от так:

files = [f for f in os.listdir('.') if re.match(r'[0-9]+.*\.jpg', f)]

Більш гнучка, але, як зазначаєте, менш ефективна.

Не ускладнювати:

import os
relevant_path = "[path to folder]"
included_extensions = ['jpg','jpeg', 'bmp', 'png', 'gif']
file_names = [fn for fn in os.listdir(relevant_path)
              if any(fn.endswith(ext) for ext in included_extensions)]

Я вважаю за краще цю форму розуміння списку, оскільки вона добре читається англійською мовою.

Я читаю четвертий рядок як: Для кожного fn в os.listdir для мого шляху, дайте мені лише ті, які відповідають будь-якому з моїх включених розширень.

Новачкам-програмістам-пітонам може бути важко реально звикнути до використання списків розуміння для фільтрації, і це може мати деяку накладну пам'ять для дуже великих наборів даних, але для перерахування каталогу та інших простих завдань фільтрації рядків розуміння списку призводить до більш чистого документальний код.

Єдине, що стосується цієї конструкції, це те, що вона не захищає вас від помилки при передачі рядка замість списку. Наприклад, якщо ви випадково перетворили рядок у список і в кінцевому підсумку перевіряєте всі символи рядка, ви можете в результаті отримати неправдиві позитиви.

Але краще мати проблему, яку легко виправити, ніж важко зрозуміти рішення.

Ще один варіант:

>>> import os, fnmatch
>>> fnmatch.filter(os.listdir('.'), '*.py')
['manage.py']

https://docs.python.org/3/library/fnmatch.html

Фільтр з globмодулем:

Імпортувати глобус

import glob

Дикі картки:

files=glob.glob("data/*")
print(files)

Out:

['data/ks_10000_0', 'data/ks_1000_0', 'data/ks_100_0', 'data/ks_100_1',
'data/ks_100_2', 'data/ks_106_0', 'data/ks_19_0', 'data/ks_200_0', 'data/ks_200_1', 
'data/ks_300_0', 'data/ks_30_0', 'data/ks_400_0', 'data/ks_40_0', 'data/ks_45_0', 
'data/ks_4_0', 'data/ks_500_0', 'data/ks_50_0', 'data/ks_50_1', 'data/ks_60_0', 
'data/ks_82_0', 'data/ks_lecture_dp_1', 'data/ks_lecture_dp_2']

Розширення Fiter .txt:

files = glob.glob("/home/ach/*/*.txt")

Одиничний персонаж

glob.glob("/home/ach/file?.txt")

Діапазони чисел

glob.glob("/home/ach/*[0-9]*")

Діапазон алфавіту

glob.glob("/home/ach/[a-c]*")

Попередній код

import glob
import fnmatch
import pathlib
import os

pattern = '*.py'
path = '.'

Рішення 1 - використовувати "глобус"

# lookup in current dir
glob.glob(pattern)

In [2]: glob.glob(pattern)
Out[2]: ['wsgi.py', 'manage.py', 'tasks.py']

Рішення 2 - використовувати "os" + "fnmatch"

Варіант 2.1 - Пошук у поточному режимі

# lookup in current dir
fnmatch.filter(os.listdir(path), pattern)

In [3]: fnmatch.filter(os.listdir(path), pattern)
Out[3]: ['wsgi.py', 'manage.py', 'tasks.py']

Варіант 2.2 - пошук рекурсивного

# lookup recursive
for dirpath, dirnames, filenames in os.walk(path):

    if not filenames:
        continue

    pythonic_files = fnmatch.filter(filenames, pattern)
    if pythonic_files:
        for file in pythonic_files:
            print('{}/{}'.format(dirpath, file))

Результат

./wsgi.py
./manage.py
./tasks.py
./temp/temp.py
./apps/diaries/urls.py
./apps/diaries/signals.py
./apps/diaries/actions.py
./apps/diaries/querysets.py
./apps/library/tests/test_forms.py
./apps/library/migrations/0001_initial.py
./apps/polls/views.py
./apps/polls/formsets.py
./apps/polls/reports.py
./apps/polls/admin.py

Рішення 3 - використовувати "pathlib"

# lookup in current dir
path_ = pathlib.Path('.')
tuple(path_.glob(pattern))

# lookup recursive
tuple(path_.rglob(pattern))

Примітки:

1. Тестований на Python 3.4
2. Модуль "pathlib" був доданий лише в Python 3.4
3. Python 3.5 додав функцію для рекурсивного пошуку за допомогою glob.glob https://docs.python.org/3.5/library/glob.html#glob.glob . Оскільки моя машина встановлена ​​з Python 3.4, я цього не перевіряв.

використовуйте os.walk для рекурсивного списку ваших файлів

import os
root = "/home"
pattern = "145992"
alist_filter = ['jpg','bmp','png','gif'] 
path=os.path.join(root,"mydir_to_scan")
for r,d,f in os.walk(path):
    for file in f:
        if file[-3:] in alist_filter and pattern in file:
            print os.path.join(root,file)

import os

dir="/path/to/dir"
[x[0]+"/"+f for x in os.walk(dir) for f in x[2] if f.endswith(".jpg")]

Це дасть вам список файлів jpg з їх повним шляхом. Ви можете замінити x[0]+"/"+fз fтільки за імена файлів. Ви також можете замінити f.endswith(".jpg")будь-яку умову рядка.

Вам також може сподобатися підхід більш високого рівня (я реалізований і упакований як findtools ):

from findtools.find_files import (find_files, Match)


# Recursively find all *.txt files in **/home/**
txt_files_pattern = Match(filetype='f', name='*.txt')
found_files = find_files(path='/home', match=txt_files_pattern)

for found_file in found_files:
    print found_file

можна встановити за допомогою

pip install findtools

Імена файлів із розширеннями "jpg" та "png" у "шлях / до / зображень":

import os
accepted_extensions = ["jpg", "png"]
filenames = [fn for fn in os.listdir("path/to/images") if fn.split(".")[-1] in accepted_extensions]

Ви можете використовувати pathlib, який доступний у стандартній бібліотеці Python 3.4 і вище.

from pathlib import Path

files = [f for f in Path.cwd().iterdir() if f.match("145592*.jpg")]

Ви можете визначити шаблон і перевірити його. Тут я взяв як початковий, так і кінцевий зразок і шукав їх у імені файлу. ФАЙЛИ містить список усіх файлів у каталозі.

import os
PATTERN_START = "145592"
PATTERN_END = ".jpg"
CURRENT_DIR = os.path.dirname(os.path.realpath(__file__))
for r,d,FILES in os.walk(CURRENT_DIR):
    for FILE in FILES:
        if PATTERN_START in FILE and PATTERN_END in FILE:
            print FILE

Як щодо str.split ()? Нічого не імпортувати.

import os

image_names = [f for f in os.listdir(path) if len(f.split('.jpg')) == 2]

Ви можете використовувати subprocess.check_ouput () як

import subprocess

list_files = subprocess.check_output("ls 145992*.jpg", shell=True) 

Звичайно, рядок між цитатами може бути будь-чим, що ви хочете виконати в оболонці, і зберегти висновок.