import os


def get_files(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(filename)  # =>吧一串字符串组合成路径
    return res

import os


def get_files(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(filename)  # =>吧一串字符串组合成路径
    return res

# 查找根目录，文件后缀 # =>当前根,根下目录,目录下的文件 # =>文件名,文件后缀 # =>吧一串字符串组合成路径