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