1. Home
  2. Python
  3. Python获取目录下所有文件

Python获取目录下所有文件

再我们使用python获取目录下的文件时，
有时需要获取特定类型的所有文件，
这里记录一个获取特定类型文件的函数：

# 获取目录下的文件
def print_all_file_path(path, suffix=''):
    url = []
    if isinstance(suffix, list) is True:
        suffix = "(" + "|".join(suffix) + ")"
    for cur_dir, sub_dir, included_file in os.walk(path):
        if included_file:
            for file in included_file:
                if re.search(suffix, file):
                    fileUrl = cur_dir + "/" + file
                    url.append(fileUrl)
    return url

使用方法：

print_all_file_path('./', '.jpg')
print_all_file_path('./', ['.jpg', '.png'])

本文为冯奎原创文章，转载无需和我联系，但请注明来自冯奎博客fengkui.net

冯奎博客
请先登录后发表评论
  • latest comments
  • 总共0条评论