1. 程式人生 > >python遍歷並輸出該資料夾與其子目錄下所有後綴為x的檔案

python遍歷並輸出該資料夾與其子目錄下所有後綴為x的檔案

path就是需要遍歷的父目錄
rule為輸出遍歷檔案的字尾
改一下變數名就闊以直接用了

# coding:utf-8
import os

def get_files(path='D:\zyt\\azyt\sfx', rule=".sfx"):
    all = []
    for fpathe,dirs,fs in os.walk(path):   # os.walk獲取所有的目錄
        for f in fs:
            filename = os.path.join(fpathe,f)
            if filename.endswith(rule):  # 判斷是否是".sfx"結尾
                all.append(filename)
    return all

if __name__ == "__main__":
    b = get_files(r"D:\zyt\\azyt\sfx")
    for i in b:
        print i