📌  相关文章
📜  在Python中的目录及其子目录中查找具有特定扩展名的文件(1)

📅  最后修改于: 2023-12-03 14:51:21.384000             🧑  作者: Mango

在Python中的目录及其子目录中查找具有特定扩展名的文件

在Python中,我们可以使用os模块和os.walk()函数来递归遍历目录及其子目录,查找具有特定扩展名的文件。

以下是一个示例代码,它可以查找指定目录下所有的.py文件:

import os

def find_files(directory, extension):
    # 遍历目录及其子目录
    for root, dirs, files in os.walk(directory):
        for file in files:
            # 判断文件是否具有指定的扩展名
            if file.endswith(extension):
                print(os.path.join(root, file))

在这个示例中,我们定义了一个find_files()函数,它接受两个参数:要查找的目录和扩展名。

我们使用os.walk()函数来遍历目录及其子目录。对于每个找到的文件,我们使用endswith()函数来检查它们是否具有指定的扩展名。如果是,我们使用os.path.join()函数构建文件的全路径并打印出来。

你可以调用这个函数来查找指定目录下所有的.py文件:

find_files('/path/to/directory', '.py')

你也可以修改这个函数来查找其他扩展名的文件:

find_files('/path/to/directory', '.txt')
find_files('/path/to/directory', '.csv')

总结

通过上面的示例,你学习了如何在Python中递归遍历目录及其子目录,查找具有特定扩展名的文件。你可以根据自己的需求修改代码以查找其他文件类型。