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文件