Python基础练习：函数应用之获取制定后缀文件名的文件

 

 

获取指定后缀名的文件：

1.导入 os

2.定义函数

import os

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

reg = find_fl('.','md') 
print(reg)                                 # 返回所有目录下的md文件