Получение отфильтрованного списка файлов в каталоге

Я пытаюсь получить список файлов в каталоге с помощью Python, но мне не нужен список всех файлов.

то, что я по существу хочу, - это возможность сделать что-то вроде следующего, но используя Python и не выполняя ls.

ls 145592*.jpg

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

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

10 ответов



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_extenstions = ['jpg', 'bmp', 'png', 'gif']
file_names = [fn for fn in os.listdir(relevant_path)
              if any(fn.endswith(ext) for ext in included_extensions)]

Я предпочитаю эту форму понимания списка, потому что она хорошо читается на английском языке.

Я прочитал четвертую строку как: Для каждого НС в ОС.listdir для моего пути, Дайте мне только те, которые соответствуют любому из моих включенных расширений.

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

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

но лучше иметь проблему, которую легко исправить, чем решение, которое трудно понять.


другой вариант:

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

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


использовать ОС.перейдите к рекурсивному списку файлов

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 glob
import fnmatch
import pathlib
import os

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

Решение 1 - используйте "glob"

# 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 - поиск в текущем dir

# 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.шарик https://docs.python.org/3.5/library/glob.html#glob.glob. Поскольку моя машина установлена с Python 3.4, я не тестировал это.

вам также может понравиться подход более высокого уровня (я реализовал и упаковал как 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

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") С любым условием строки вы хотите.


имена файлов с расширениями" jpg "и" png " в "path/to / images":

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

вы можете использовать подпроцесс.check_ouput() в качестве

import subprocess

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

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