获取后缀名的文件

时间:2022-07-25
本文章向大家介绍获取后缀名的文件，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。
import os

def find_file(work_dir,extension='jpg'):
    lst=[]
    for filename in os.listdir(work_dir):
        print(filename)
        splits=os.path.splitext(filename)
        ext=splits[1] #拿到扩展名
        if ext=='.'+extension:
            lst.append(filename)
    return lst

r=find_file('.','py')
print(r) #返回所有目录下的md文件

随机文章
本站知识点必读
最近更新