Удалить файлы с разными именами в Python

Question

Я пытаюсь удалить несколько файлов в каталоге.

До сих пор у меня есть этот код:

for filename in glob.glob("buffer*"): 
     os.remove(filename) 
    for filename in glob.glob("grid*"): 
     os.remove(filename) 
    for filename in glob.glob("OSMroads*"): 
     os.remove(filename) 
    for filename in glob.glob("newCostSurface*"): 
     os.remove(filename) 
    for filename in glob.glob("standsLine*"): 
     os.remove(filename) 
    for filename in glob.glob("standsReprojected*"): 
     os.remove(filename) 

Есть ли способ сделать это более эффективным?

Answer 1

Doing 6 отдельных glob вызовы, конечно, итерацию объект каталога 6 раз.

К счастью, практически на любой платформе он, вероятно, в конечном итоге будет кэширован после первого раза. Если ваш каталог не является абсолютно гигантским, это не будет заметной проблемой.

Но так как вы явно задали вопрос об эффективности, вы можете, очевидно, повторить и фильтровать результаты. Самый простой способ сделать это - fnmatch. Все, что делает glob, вызывает listdir, а затем fnmatch на каждый результат; Вы можете сделать то же самое с несколькими fnmatch вызовов:

for filename in os.listdir('.'): 
    if fnmatch.fnmatch(filename, 'buffer*'): 
     os.remove(filename) 
    # etc. 

И, конечно, вы можете упростить это точно так же, как partofthething упрощена существующий код:

for filename in os.listdir('.'): 
    for pattern in ['buffer*', 'grid*', 'OSMroads*', 
        'newCostSurface*','standsLine*', 'standsReprojected*']: 
     if fnmatch.fnmatch(filename, pattern): 
      os.remove(filename) 

Или:

for filename in os.listdir('.'): 
    if any(fnmatch.fnmatch(filename, pattern) 
      for pattern in ['buffer*', 'grid*', 'OSMroads*', 
          'newCostSurface*','standsLine*', 'standsReprojected*']): 
     os.remove(filename) 

Если вам действительно нужно выжать еще одну крошечную долю процента производительности, вы можете использовать fnmatch.translate для преобразования каждого шаблона в регулярное выражение, а затем слить r egexps в чередование и скомпилировать его, а затем применить этот объект регулярного выражения к каждому имени файла. Но время CPU для fnmatch по сравнению с временем ввода-вывода для чтения объектов каталога, вероятно, настолько мало, что улучшение даже не измеримо.

Answer 2

Мне нравится использовать списки, так что я не повторять код, например:

for pattern in ['buffer*','grid*','OSMroads*','newCostSurface*','standsLine*' 
       'standsReprojected*']: 
    for filename in glob.glob(pattern): 
     os.remove(filename)