Obtenir une liste filtrée de fichiers dans un répertoire

Question

J'essaie d'obtenir une liste de fichiers dans un répertoire en utilisant Python, mais je ne veux pas une liste de TOUS les fichiers.

Ce que je veux essentiellement, c'est pouvoir faire quelque chose comme ce qui suit, mais en utilisant Python et sans exécuter ls.

ls 145592*.jpg

S'il n'y a pas de méthode intégrée pour cela, je pense actuellement à écrire une boucle for pour itérer à travers les résultats d'un os.listdir() et d'ajouter tous les fichiers correspondants à une nouvelle liste.

Cependant, il y a beaucoup de fichiers dans ce répertoire et j'espère donc qu'il existe une méthode plus efficace (ou une méthode intégrée).

Answer 1

import glob

jpgFilenamesList = glob.glob('145592*.jpg')

Ver glob dans la documentation python

Answer 2

glob.glob() est définitivement la façon de faire (selon Ignacio). Cependant, si vous avez besoin d'une correspondance plus compliquée, vous pouvez le faire avec une compréhension de liste et re.match() quelque chose comme ça :

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

Plus flexible, mais comme vous le notez, moins efficace.

Answer 3

Restez simple :

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)]

Je préfère cette forme de compréhension de liste car elle se lit bien en anglais.

Je lis la quatrième ligne comme : Pour chaque fn dans os.listdir pour mon chemin, donnez-moi seulement ceux qui correspondent à l'une de mes extensions incluses.

Il peut être difficile pour les programmeurs python novices de s'habituer à l'utilisation des compréhensions de listes pour le filtrage, et il peut y avoir une surcharge de mémoire pour les très grands ensembles de données, mais pour lister un répertoire et d'autres tâches simples de filtrage de chaînes, les compréhensions de listes conduisent à un code plus propre et documentable.

Le seul inconvénient de cette conception est qu'elle ne vous protège pas contre l'erreur de passer une chaîne de caractères au lieu d'une liste. Par exemple, si vous convertissez accidentellement une chaîne de caractères en liste et que vous vous retrouvez à vérifier tous les caractères d'une chaîne de caractères, vous risquez d'obtenir un grand nombre de faux positifs.

Mais il est préférable d'avoir un problème facile à résoudre qu'une solution difficile à comprendre.

Answer 4

Utiliser os.walk pour lister récursivement vos fichiers

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)

Answer 5

Vous pourriez aussi aimer une approche plus haut niveau (j'ai implémenté et emballé en tant que 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

peut être installé avec

pip install findtools