[python日常]获取指定文件夹下，指定后缀的文件
import os
def get_files_by_suffix(folder_path, suffix):
"""
获取指定文件夹下所有指定后缀的文件列表
:param folder_path: 文件夹路径
:param suffix: 文件后缀
:return: 文件列表
"""
return [os.path.join(folder_path, file) for file in os.listdir(folder_path) if file.endswith(suffix)]
# 使用示例
folder = '/path/to/your/folder' # 替换为你的文件夹路径
suffix = '.txt' # 替换为你想要的文件后缀
files = get_files_by_suffix(folder, suffix)
print(files)这段代码定义了一个get_files_by_suffix函数，它接受文件夹路径和文件后缀作为参数，返回该文件夹下所有指定后缀的文件列表。使用os.listdir获取文件夹内的所有文件，然后使用列表推导式过滤出指定后缀的文件，最后返回这些文件的完整路径。使用时只需要替换folder和suffix变量的值即可。
评论已关闭