Question

我有这段代码可以找到具有该扩展名的文件并将其删除。

如何在两个目录中搜索？ out和out2

import os

out = path
out2 = path2

test = os.listdir(out)
for item in test:
    if item.endswith(".cpg"):
        os.remove(os.path.join(out, item))

不用第二个目录再次输入吗？

注意：所有这些都是for循环的一部分。

Answer 1

您可以使其成为函数：

import os

out = path
out2 = path2

directory_search(out)
directory_search(out2)

def directory_search(out):
    test = os.listdir(out)
    for item in test:
        if item.endswith(".cpg"):
            os.remove(os.path.join(out, item))

或者您列出一个列表并遍历：

import os

list_of_out = [path, path2]
for out in list_of_out:
    test = os.listdir(out)
    for item in test:
        if item.endswith(".cpg"):
            os.remove(os.path.join(out, item))

或者我想如果您真的想做的话，可以创建一个带有列表的函数：

import os

out = [path, path2]
directory_search(out)

def directory_search(list_of_out):
    for out in list_of_out:
            test = os.listdir(out)
            for item in test:
                if item.endswith(".cpg"):
                    os.remove(os.path.join(out, item))

Answer 2

转到目录级别，例如C：/，d：/：在目录级别进行循环遍历所有文件和文件夹，然后在每个文件夹中遍历每个文件和文件夹。您可以将LinkedList / Array用于要扫描的文件夹，并一一扫描它们

检查多个目录是否具有某些扩展名的文件？

2 个答案: