Python：使用yield函数遍历目录

Python：使用yield函数遍历目录

如果你正在处理大量数据或文件，Python 的 yield 函数可以帮助你轻松实现目录遍历和文件操作。本文将为您介绍使用 yield 函数遍历目录的方法。

一、yield 函数的作用

在 Python 中，yield 函数是一个特殊的生成器函数，它可以将生成器对象返回给调用者，而不是将所有结果一次性返回。这样可以节省内存，提高程序的执行效率。

二、使用 os 模块获取目录信息

os 模块是 Python 中用于处理操作系统相关任务的标准库之一。通过导入 os 模块，我们可以使用其中的函数来获取目录中的信息。

import os

def walk_dir(dir_path):
    for name in os.