python遍歷讀取文件下的所有指定類型的文件

#!/usr/bin/env python3
# -*- coding:utf-8 -*-

# 1. for-in dir/subdir to get the filesname
# 2. splitext filename to filter

import os

def getFiles(dir, suffix): # 查找根目錄，文件後綴 
    res = []
    for root, directory, files in os.walk(dir):  # =>當前根,根下目錄,目錄下的文件
        for filename in files:
            name, suf = os.path.splitext(filename) # =>文件名,文件後綴
            if suf == suffix:
                res.append(os.path.join(root, filename)) # =>吧一串字符串組合成路徑
    return res

for file in getFiles("./", '.py'):  # =>查找以.py結尾的文件
    print(file)