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結尾的文件名的列表