Question

from keras_applications.imagenet_utils import _obtain_input_shape

在上面的示例dir中，我需要使用glob模式执行递归删除。

DirA
  file1.txt
  file2.conf
  DirB
      file3.txt
      file4.sh
      file5.sh
      DirC
          file6.bat
          file7.txt

使用上述模式，我需要删除所有目录中所有带有* .txt和* .sh格式的文件

Answer 1

import os, glob

pattern = ['*.txt', '*.sh']

for p in pattern:
    [os.remove(x) for x in glob.iglob('DirA/**/' + p, recursive=True)]

如果需要，可以使用列表推导来完成此任务。 Cya！

Answer 2

import os
import glob

for filename in glob.iglob('DirA/**/*.txt', recursive=True):
    os.remove(filename)
for filename in glob.iglob('DirA/**/*.sh', recursive=True):
    os.remove(filename)

这应该可以递归删除目录中的所有txt和sh文件。

或者如果您想使用模式指定数组：

import os
import glob


def removeAll(pathToDir, patterns):
    for pattern in patterns:
        for filename in glob.iglob(pathToDir + '/**/' + pattern, recursive=True):
            os.remove(filename)


patterns = ['*.txt', '*.sh']

removeAll('DirA', patterns)

Answer 3

您可以改用os.walk，这样您只需要遍历目录一次。使用glob时，您必须遍历两次，因为您要查找两种模式。

import os
for root, _, files in os.walk('DirA'):
    for file in files:
        if any(file.endswith(ext) for ext in pattern):
            os.remove(os.path.join(root, file))

如何在使用python的目录中使用glob模式执行递归删除？

3 个答案: