Question

我有一个目录，其他文件夹中可以有多个文件夹，其中包含txt文件。我想返回包含至少一个.txt文件的所有目录的列表。

我正在尝试以下递归方法，但它不起作用：

def getDirectoryList(dir):
    directoryList = []

    # return nothing if dir is a file
    if not os.path.isdir(dir):
        return []

    # add dir to directorylist if it contains .txt files
    if len([file for file in os.listdir(dir) if file.endswith('.txt')])>0:
        directoryList.append(dir)
    for d in os.listdir(dir):
        for x in getDirectoryList(d):
            directoryList.append[x]

    return directoryList

Answer 1

def getDirectoryList(path):
    directoryList = []

    #return nothing if path is a file
    if os.path.isfile(path):
        return []

    #add dir to directorylist if it contains .txt files
    if len([f for f in os.listdir(path) if f.endswith('.txt')])>0:
        directoryList.append(path)

    for d in os.listdir(path):
        new_path = os.path.join(path, d)
        if os.path.isdir(new_path):
            directoryList += getDirectoryList(new_path)

    return directoryList

这是有效的代码。重要的区别是if“os.path.isdir（os.path.join（path，d））：”在递归调用之前检查并添加路径和d，因为os.listdir（）给出了名称，而不是路径< / p>

Python查找包含.txt文件的所有子目录

1 个答案: