python 获取路径下指定后缀的文件

在Python中，我们可以使用`os`模块来获取路径下指定后缀的文件。具体步骤如下：

1. 首先，导入`os`模块：`import os`
2. 创建一个空列表来存储符合条件的文件路径：`files = []`
3. 使用`os.listdir()`函数获取指定路径下的所有文件和文件夹：`items = os.listdir(path)`
4. 遍历路径下的所有项：
- 对于每个项，使用`os.path.join()`函数将路径和项名连接起来，得到完整的文件或文件夹路径：`item_path = os.path.join(path, item)`
- 使用`os.path.isdir()`函数判断该路径是否是一个文件夹：
- 如果是文件夹，递归调用步骤4来处理该文件夹下的所有项
- 如果不是文件夹，使用`os.path.splitext()`函数获取文件的后缀名，并判断是否符合指定后缀：
- 如果符合指定后缀，将该文件路径添加到`files`列表中：`files.append(item_path)`
5. 返回`files`列表，其中包含路径下所有符合指定后缀的文件。

下面是一个示例函数来实现上述过程：

import os

def get_files_with_extension(path, extension):
    files = []
    items = os.listdir(path)
    for item in items:
        item_path = os.path.join(path, item)
        if os.path.isdir(item_path):
            files += get_files_with_extension(item_path, extension)
        else:
            if os.path.splitext(item_path)[1] == extension:
                files.append(item_path)
    return files

你可以调用这个函数，并将路径和指定后缀作为参数传递，函数将返回一个包含指定后缀的文件路径列表。