Рекурсивно разворачивать и искать шаблон для определенных подкаталогов

Question

Я ищу возможность поиска определенных подкаталогов в python.

Например структура каталогов, как это:

some_files/ 
    common/ 
    2009/ 
    2010/ 
    2011/ 
    ... 

Я только хочу, чтобы искать в подкаталогах, которые начинаются с 2, так что это должно быть что-то вроде «some_files/2 *». Я думаю, что это должно быть возможно, используя glob.glob и os.walk(), но я не могу заставить его работать.

Сейчас я использую:

files = [os.path.join(dirpath, f) 
       for dirpath, dirnames, files in os.walk(d) 
       for f in files if f.endswith(ext)] 

, но это не соответствует конкретным потребностям.

Может кто-то помочь мне, будет очень признателен!

Answer 1

Вы можете использовать Glob с dirpath, чтобы найти соответствующие каталоги:

from glob import iglob 
import os 

files = [] 
ext = "py" 
for dirpath, dirnames, file in os.walk(path): 
    match = next(iglob(os.path.join(dirpath, "2*")),"") 
    if match: 
     files.extend(iglob(os.path.join(match,"*.{}".format(ext)))) 
print(files) 

Или, если вы действительно хотите список Комп:

files = [f for dirpath, dirnames, file in os.walk(path) for f in 
     iglob(os.path.join(next(iglob(os.path.join(dirpath, "2*")), 
           '\\\\'), "*.{}".format(ext)))] 
print(files) 

Answer 2

Я хотел бы сделать это, как это, используя pathlib (, который является теперь частью Python3 станд Lib):

from pathlib import Path 

for subpath in Path().glob("2*): 
    for file in subpath.glob("*.ext"): 
     # ... 

Update:pathlib также доступна для Python 2.x (это было с обратной связью и опубликованы до Python Package Index). Просто:

$ pip install pathlib