写一个递归函数列出parts.txt的所有路径

Question

编写一个函数 list_files_recursive 来 returns 所有 parts.txt 文件的路径列表，而不使用 os 模块的步行生成器。相反，函数应该使用递归。输入将是目录名称。这是我到目前为止的代码，我认为它基本上是正确的，但发生的事情是输出不是一个完整的列表？

def list_files_recursive(top_dir):
    rec_list_files = []
    list_dir = os.listdir(top_dir)
    for item in list_dir:
        item_path = os.path.join(top_dir, item)
        if os.path.isdir(item_path):
            list_files_recursive(item_path)
        else:
            if os.path.basename(item_path) == 'parts.txt': 
                rec_list_files.append(os.path.join(item_path))
    print(rec_list_files)
    return rec_list_files

这是我得到的输出的一部分（来自 print 语句）：

['CarItems/Honda/Accord/1996/parts.txt']
[]
['CarItems/Honda/Odyssey/2000/parts.txt']
['CarItems/Honda/Odyssey/2002/parts.txt']
[]

所以问题是它不是一个列表，里面有空列表。我不太清楚为什么这不起作用，并且已经尝试了一切来解决它。非常感谢任何帮助！

Answer 1

这非常接近，但问题是 list_files_recursive 的子调用不会将结果传回给父调用。一种方法是将每个子调用的所有列表连接在一起，或者在整个调用链中传递对单个列表的引用。

请注意，在 rec_list_files.append(os.path.join(item_path)) 中，os.path.join 中只有一个参数是没有意义的。 print(rec_list_files) 应作为 side effect 省略，这会使输出难以解释——仅在调用者中打印。此外，

else:
    if ... :

在这里可以更清楚地写成 elif: 因为它们在逻辑上是等价的。尽可能减少条件语句的嵌套总是一个好主意。

这是通过扩展父列表起作用的方法：

import os 

def list_files_recursive(top_dir):
    files = []

    for item in os.listdir(top_dir):
        item_path = os.path.join(top_dir, item)

        if os.path.isdir(item_path):
            files.extend(list_files_recursive(item_path)) 
            #     ^^^^^^ add child results to parent
        elif os.path.basename(item_path) == "parts.txt": 
            files.append(item_path)

    return files

if __name__ == "__main__":
    print(list_files_recursive("foo"))

或者通过调用树传递结果列表：

import os 

def list_files_recursive(top_dir, files=[]):
    for item in os.listdir(top_dir):
        item_path = os.path.join(top_dir, item)

        if os.path.isdir(item_path):
            list_files_recursive(item_path, files)
            #                               ^^^^^ pass our result list recursively
        elif os.path.basename(item_path) == "parts.txt": 
            files.append(item_path)

    return files

if __name__ == "__main__":
    print(list_files_recursive("foo"))

这些函数的一个主要问题是它们仅适用于查找精确命名为 parts.txt 的文件，因为该字符串文字是 hard coded。这使得它除了直接目的之外几乎没有任何用处。我们应该添加一个参数，让调用者指定他们要搜索的目标文件，使函数通用。

另一个问题是该函数没有按照其名称声称的那样进行操作：list_files_recursive 实际上应该被称为 find_file_recursive，或者，由于硬编码字符串，find_parts_txt_recursive。

除此之外，该函数是转换为 generator 函数的有力候选者，这是遍历的常见 Python 习惯用法，特别是在子目录可能包含大量数据的情况下一次将所有内容保存在内存中会很昂贵。生成器还允许灵活地使用函数在第一次匹配后取消搜索，进一步增强其（重）可用性。

yield 关键字还使函数代码本身非常干净——我们可以避免完全保留结果数据结构的问题，而只是按需触发结果项。

我会这样写：

import os 

def find_file_recursive(top_dir, target):
    for item in os.listdir(top_dir):
        item_path = os.path.join(top_dir, item)

        if os.path.isdir(item_path):
            yield from find_file_recursive(item_path, target)
        elif os.path.basename(item_path) == target:
            yield item_path

if __name__ == "__main__":
    print(list(find_file_recursive("foo", "parts.txt")))

写一个递归函数列出parts.txt的所有路径

Write a recursive function to list all paths of parts.txt

python-3.x

list

directory

os.walk