python根据文件名批量搜索文件

下面我会给出一个详细的 tutorial，教你如何使用 Python 根据文件名批量搜索文件。

步骤1：导入必要的模块

在开始之前我们需要导入两个非常重要的模块：os 和 fnmatch。os 库为操作系统提供了接口函数，fnmatch 则提供了 Unix shell 样式的通配符，通过这两个模块的结合能力我们可以实现在特定文件夹内根据文件名批量搜索文件。

import os
import fnmatch

步骤2：编写函数

我们创建一个名为 find_files 的函数，这个函数接收2个参数：root 和 pattern，其中 root 为需要搜索的文件夹名，pattern 则为搜索的文件名。这个函数返回一个列表形式的结果，数组内包含了查询到的文件名。

def find_files(root, pattern):
    for path, dirs, files in os.walk(root):
        for filename in fnmatch.filter(files, pattern):
            yield os.path.join(path, filename)

在这个函数中，os.walk 函数获取了一个由目录名称和子目录组成的元组，以及一个由文件名组成的列表。我们使用 fnmatch.filter 函数来查找所有与给定模式匹配的文件名。然后使用 os.path.join 函数来链接这些文件名，从而返回一个包含所有匹配文件路径的生成器。

步骤3：调用函数并输出结果

现在我们可以使用我们的函数来搜索名为 "test.py" 的 Python 文件，代码如下所示：

python_files = list(find_files(".", "*.py"))
for file in python_files:
    print(file)

作为另一个示例，假设我们要查找名为 "example.txt" 的所有文本文件。我们可以使用以下代码：

text_files = list(find_files(".", "example*.txt"))
for file in text_files:
    print(file)

这里的 pattern 是 "example*.txt"，它会匹配所有以 "example" 开头以 .txt 结尾的文件，例如 "example.txt", "example1.txt", "example_123.txt" 等。

这就是使用 Python 根据文件名批量搜索文件的完整攻略。