遍历查找含特定字符的文件并排序2

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 == '.end' or suf == '.dat':
                # res.append(os.path.join(root, filename))  # =>把一串字符串组合成路径
                if suffix in filename:
                    res.append(filename)
    return res
    
    
def getFileList(dir_name, suffix):
    key_value = {}
    for file in getFiles(dir_name,suffix):
        num = int(os.path.splitext(file)[0].split('_', 9)[9])
        key_value[num] = file
        
    list_file = []
    for i in sorted(key_value):
        list_file.append(dir_name + key_value[i])
        print(key_value[i])
        
    return list_file