Liczba plików z pewnym rozszerzeniem w Pythonie

Question

Jestem całkiem nowy w Pythonie i staram się znaleźć najbardziej efektywny sposób zliczania plików .TIF w określonym podkatalogu.

Robi kilka wyszukiwania, znalazłem jeden przykład (nie testowałem), który twierdził, że liczą się wszystkie pliki w katalogu:

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

Jest to dobre rozwiązanie, ale trzeba liczyć tylko pliki TIF . Mój katalog będzie zawierał inne typy plików, ale chcę tylko liczyć TIF-y.

Obecnie używam następujący kod:

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

To działa dobrze, ale pętli wydaje się nadmierna/drogie dla mnie. W jaki sposób można to zrobić bardziej efektywnie?

Dzięki.

Answer 1

Coś musi powtórzyć wszystkie pliki w katalogu i sprawdzić każdą nazwę pliku - niezależnie od tego, czy jest to kod, czy biblioteka. Bez względu na to, jakie konkretne rozwiązanie, wszystkie będą miały mniej więcej ten sam koszt.

Jeśli uważasz, że to zbyt dużo kodu, a jeśli nie faktycznie trzeba szukać podkatalogi rekurencyjnie, można użyć modułu glob:

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

Answer 2

Twój kod jest w porządku.

Tak, będziesz musiał przechwycić te pliki, aby odfiltrować pliki .tif, ale pętla nad małą tablicą w pamięci jest pomijalna w porównaniu do pracy skanowania katalogu plików, aby znaleźć te pliki w pierwsze miejsce, które i tak musisz zrobić.

Nie martwię się o optymalizację tego kodu.

Answer 3

Jeśli musisz szukać rekursywnie, lub dla niektórych Innym powodem nie chcą korzystać z modułu glob, można użyć

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

jest to „pythonowy” sposób dostosować przykład znalazłeś dla swoich celów. Ale nie będzie to znacznie szybsze ani wydajniejsze niż pętla, z której korzystasz; to po prostu naprawdę kompaktowa składnia mniej więcej to samo.

Answer 4

W tym konkretnym przypadku, jeśli nie chce rekursywnie szukać w podkatalogu, można użyć os.listdir:

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