с использованием метода os.walk для получения путей к каталогам, содержащих 2 типа файлов

Question

Я хочу перечислить все папки, содержащие docx, используя os() .walk метод в Python 2.7. Мне удалось сделать это с помощью кода, написанного ниже, но я хочу знать, можно ли ограничить этот список, чтобы показывать только папки, содержащие ровно два определенных типа файлов (например, «docx» и «pdf»)?

import os 
import walk 

a = open("output.txt", "w") 
for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'): 
    for filename in files: 
     if filename.endswith(('.docx')): 
      f = os.path.join(path, filename) 
      a.write(str(f) + os.linesep) 

Answer 1

Просто пропустите каталоги, в которых у вас нет хотя бы этих двух расширений; в-справочник списки файлов ограничены, так что это дешево использовать any() для проверки конкретных расширений:

for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'): 
    if not (any(f.endswith('.pdf') for f in files) and 
      any(f.endswith('.docx') for f in files)): 
     # no PDF or Word files here, skip 
     continue 
    # directory contains *both* PDF and Word documets 

Когда список расширений для тестирования для становится немного больше, вы можете просто создать набор всех доступных расширения:

for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'): 
    extensions = {os.path.splitext(f)[-1] for f in files} 
    if not extensions >= {'.pdf', '.docx', '.odt', '.wpf'}: 
     # directory doesn't contain *all* required file types 
     continue 

>= тесты, если множество правая является подмножеством левой (так extensions является superset of the right-hand set); так extensions должны по крайней мере содержат все для расширения имени на праве:

>>> {'.foo', '.docx', '.pdf', '.odt'} >= {'.pdf', '.docx', '.odt', '.wpf'} # missing .wpf 
False 
>>> {'.foo', '.wpf', '.docx', '.pdf', '.odt'} >= {'.pdf', '.docx', '.odt', '.wpf'} # complete 
True 

Answer 2

этом?

import os 

a = open("output.txt", "w") 
for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'): 
    docx = False 
    pdf = False 
    rest = True 
    for filename in files: 
     if filename.endswith(('.docx')): 
      docx = True 
     elif filename.endswith(('.pdf')): 
      pdf = True 
     else: 
      rest = False 
      break 
    if docx and pdf and rest: 
     f = os.path.join(path, filename) 
     a.write(str(f) + os.linesep)