### Python 实现获取当前目录下以 jpg 和 png 结尾的文件 以下是通过 `os` 模块和 `fnmatch` 模块来实现的功能代码： ```python import os import fnmatch def get_jpg_png_files(): """ 获取当前目录下所有以 .jpg 或 .png 结尾的文件。 :return: 匹配到的文件名列表 """ result = [] current_directory = "." # 当前目录 for folder_name, _, file_names in os.walk(current_directory): for file_name in file_names: if fnmatch.fnmatch(file_name, "*.jpg") or fnmatch.fnmatch(file_name, "*.png"): result.append(os.path.join(folder_name, file_name)) return result if __name__ == "__main__": files = get_jpg_png_files() print(files) ``` 此代码会遍历当前目录及其子目录中的所有文件，并筛选出扩展名为 `.jpg` 或 `.png` 的文件[^1]。 如果仅需查找当前目录而不包括其子目录，则可修改代码如下： ```python import os import fnmatch def get_jpg_png_files_in_current_dir(): """ 仅获取当前目录（不包括子目录）下所有以 .jpg 或 .png 结尾的文件。 :return: 匹配到的文件名列表 """ result = [] current_directory = "." all_files = [f for f in os.listdir(current_directory) if os.path.isfile(f)] for file_name in all_files: if fnmatch.fnmatch(file_name, "*.jpg") or fnmatch.fnmatch(file_name, "*.png"): result.append(file_name) return result if __name__ == "__main__": files = get_jpg_png_files_in_current_dir() print(files) ``` 这段代码只会在当前目录中寻找符合条件的文件，而不会深入子目录[^2]。 对于更复杂的模式匹配需求，还可以考虑使用正则表达式替代 `fnmatch.fnmatch()` 函数。