Graf Anzahl der Dateien mit bestimmter Erweiterung in Python

Question

Ich bin ziemlich neu in Python und ich versuche herauszufinden, die effizienteste Möglichkeit, die Anzahl der. TIF-Dateien in einem bestimmten Unterverzeichnis zu zählen.

einige der Suche, fand ich ein Beispiel (habe ich nicht getestet), die alle Dateien in einem Verzeichnis zu zählen behauptet:

file_count = sum((len(f) for _, _, f in os.walk(myPath))) 

Das ist in Ordnung, aber ich brauche nur TIF-Dateien zählen . Mein Verzeichnis enthält andere Dateitypen, aber ich möchte nur TIFs zählen.

Derzeit bin ich mit dem folgenden Code:

tifCounter = 0 
for root, dirs, files in os.walk(myPath): 
    for file in files:  
     if file.endswith('.tif'): 
      tifCounter += 1 

Es funktioniert gut, aber der Looping scheint mir übertrieben/teuer. Jeder Weg, dies effizienter zu tun?

Danke.

Answer 1

Irgendetwas muss über alle Dateien im Verzeichnis iterieren und jeden einzelnen Dateinamen betrachten - ob das Ihr Code oder eine Bibliotheksroutine ist. Also, egal welche spezifische Lösung, sie werden alle ungefähr die gleichen Kosten haben.

Wenn Sie denken, es ist zu viel Code, und wenn Sie tatsächlich nicht rekursiv Unterverzeichnisse suchen müssen, können Sie das glob Modul verwenden:

tifCounter = len(glob.glob1(myPath,"*.tif")) 

Answer 2

Ihr Code ist in Ordnung.

Ja, Sie müssen diese Dateien durchlaufen, um die TIF-Dateien herauszufiltern, aber das Schleifen über ein kleines In-Memory-Array ist vernachlässigbar verglichen mit dem Durchsuchen des Dateiverzeichnisses nach diesen Dateien der erste Ort, den du sowieso machen musst.

Ich würde mir keine Sorgen über die Optimierung dieses Codes machen.

Answer 3

Wenn Sie rekursiv suchen brauchen, oder für einige anderer Grund nicht will, das glob Modul verwenden, Sie

file_count = sum(len(f for f in fs if f.lower().endswith('.tif')) for _, _, fs in os.walk(myPath)) 

Dies ist die „Pythonic“ Art und Weise anzupassen, das Beispiel, das Sie für Ihre Zwecke verwenden Sie können. Aber es wird nicht wesentlich schneller oder effizienter als die Schleife, die Sie verwendet haben; es ist nur eine sehr kompakte Syntax für mehr oder weniger die gleiche Sache.

Answer 4

Für diesen speziellen Anwendungsfall, wenn Sie rekursiv nicht im Unterverzeichnis suchen mögen, können Sie os.listdir verwenden:

len([f for f in os.listdir(myPath) 
    if f.endswith('.tif') and os.path.isfile(os.path.join(myPath, f))])