使用 Python 实现文件递归遍历的三种方式

在Python编程中，遍历文件和目录是常见的任务，特别是在处理大量数据或执行自动化操作时。本文将介绍三种使用Python实现文件递归遍历的方法，包括基础方法、使用`os.walk()`以及自定义优化。 让我们从基础方法开始。在最初的示例中，我们看到有两种函数：`getallfiles()`和`getalldirfiles()`。`getallfiles()`仅遍历当前目录中的文件，而`getalldirfiles()`递归地遍历子目录。这两个函数都使用了`os.listdir()`来获取目录中的文件和子目录列表。这种实现方式虽然能工作，但并不高效，因为它需要多次调用`listdir()`，并且代码结构不够清晰。 优化后的第一种方法是合并这两个函数，只调用一次`listdir()`，通过`if~else~`判断处理文件和目录。代码如下： ```python def getallfiles(dir): """使用listdir循环遍历""" if not os.path.isdir(dir): print(dir) return dirlist = os.listdir(dir) for dirret in dirlist: fullname = dir + "\\" + dirret if os.path.isdir(fullname): getallfiles(fullname) else: print(fullname) ``` 尽管这个改进减少了一次`listdir()`调用，但仍然存在递归调用。 第二种优化方法是利用Python标准库`os`中的`os.walk()`函数。`os.walk()`是一个非常实用的工具，它可以递归地遍历目录树并返回一个生成器，每次迭代都会提供当前目录的路径（root），该目录下的子目录列表（dirs）和文件列表（files）。这样，我们可以简化代码如下： ```python def getallfilesofwalk(dir): """使用os.walk()遍历""" if not os.path.isdir(dir): print(dir) return for root, dirs, files in os.walk(dir): for file in files: print(os.path.join(root, file)) ``` `os.walk()`的优点在于其简洁性和易于理解。然而，如前所述，`os.walk()`实际上也是通过多次调用`listdir()`来实现的，它只是对输出结果进行了额外的处理，以提供更方便的接口。 `os.walk()`的源码实现通常会考虑到错误处理和权限问题，以确保即使在访问某些目录时遇到权限错误，程序仍能继续运行。例如，如果在遍历过程中遇到无法读取的目录，`os.walk()`会忽略这个异常，继续处理其他可读目录。 递归遍历文件系统时，`os.walk()`是推荐的首选方法，因为它提供了简洁的接口和良好的性能。然而，对于简单的任务，自定义的递归函数也可以满足需求，只要注意避免不必要的函数调用和提高代码效率。在实际应用中，根据项目需求和性能要求选择合适的方法。