python找出某个文件夹下某个后缀的文件

python找出某个文件夹下某个后缀的文件

import os

def find_files_with_suffix(folder_path, suffix):
    # 使用os模块获取文件夹中所有文件的路径
    all_files = os.listdir(folder_path)

    # 筛选以指定后缀名结尾的文件
    filtered_files = [file for file in all_files if file.endswith(suffix)]

    return filtered_files

该函数接受一个文件夹路径和一个后缀名作为参数，并返回一个包含所有以该后缀名结尾的文件名的列表
你可以将/path/to/folder替换为你要搜索的文件夹路径，将.txt替换为你要搜索的后缀名。例如，要查找所有以.pdf结尾的文件，你可以这样调用这个函数：

pdf_files = find_files_with_suffix("/path/to/folder", ".pdf")
print(pdf_files)

这个程序将输出一个包含所有以.pdf结尾的文件名的列表