如何使用 Python 查找文件？

在日常开发过程中，我们经常需要查找特定类型的文件。在 Python 中，我们可以使用 os 模块提供的 walk() 方法查找文件，并根据需要筛选结果。

步骤

1. 导入 os 模块

首先，在开始编写 Python 代码之前，我们需要先导入 os 模块。可通过以下代码实现：

import os

2. 定义查找函数

我们需要定义一个函数，以便在需要时调用。可以为该函数命名为 find_files()，它将接受两个参数：要搜索的目录路径和要查找的文件扩展名。

def find_files(path, file_extension):
    result = []
    # TODO: implementation
    return result

3. 遍历目录及其子目录

可以使用 os.walk() 函数遍历指定目录下的所有文件及其子目录。该函数返回一个元组，其中包含当前遍历的目录路径、该目录下的子目录列表和该目录下的文件列表。

def find_files(path, file_extension):
    result = []
    for root, dirs, files in os.walk(path):
        # TODO: implementation
    return result

4. 筛选文件

对于每个遍历到的文件，我们需要检查其是否满足我们的搜索条件。在此示例中，搜索条件就是文件扩展名是否与目标扩展名匹配。可以使用 Python 字符串的 endswith() 方法检查文件名是否以指定的扩展名结尾。

def find_files(path, file_extension):
    result = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(file_extension):
                # TODO: implementation
    return result

5. 添加匹配的文件路径到结果列表中

如果找到了一个满足条件的文件，我们需要将其路径添加到结果列表中。在 Python 中，可以使用 os.path.join() 方法将目录路径和文件名组合成一个完整的路径。

def find_files(path, file_extension):
    result = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(file_extension):
                result.append(os.path.join(root, file))
    return result

示例代码

以下是一个完整的示例代码：

import os

def find_files(path, file_extension):
    result = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(file_extension):
                result.append(os.path.join(root, file))
    return result

# Example usage
path = 'C:/Users/username/Documents'
file_extension = '.txt'
found_files = find_files(path, file_extension)
print(found_files)

在这个示例中，我们定义了一个名为 find_files 的函数，它接受两个参数：要搜索的目录路径和要查找的文件扩展名，函数使用 os.walk() 遍历目录及其子目录，并检查每个文件是否以给定的扩展名结尾，如果满足条件，将文件路径添加到结果列表中，函数返回结果列表。

推荐问题

• 如何在 Python 中读取 CSV 文件？
• 如何在 Python 中实现单元测试？
• 如何在 Python 中生成随机数？

如果您有任何问题或建议，请在下面的评论区留言，也欢迎关注、点赞和分享！感谢您的阅读！

引用图片：