python 遍历指定文件夹指定类型文件

import os
path="d:\\python37"
filetype=".pdf"
#遍历包括子文件夹
def get_filename(path,filetype):
    filetype1=filetype.upper()
    #print(filetype)
    name =[]
    final_name = []
    for root,dirs,files in os.walk(path):
        for i in files:
            #print(i)
            if i.endswith(filetype) or i.endswith(filetype1):
                final_name.append(i)
    return final_name
#遍历当前文件夹，不包括子文件夹
def get_filenamea(path,filetype):
    filetype1=filetype.upper()
    #print(filetype)
    name =[]
    final_name = []
    for files in os.listdir(path):
        if files.endswith(filetype) or files.endswith(filetype1):
            final_name.append(files)
    return final_name

print(get_filenamea(path,filetype))
print(get_filename(path,filetype))