python 讀取指定文件夾中的指定文件類型的文件名

import numpy as np
import os

path = 'F:\\wenjian'#指定文件所在路徑
filetype ='.csv'#指定文件類型
def get_filename(path,filetype):
    name =[]
    final_name = []
    for root,dirs,files in os.walk(path):
        for i in files:
            if filetype in i:
                name.append(i.replace(filetype,''))#生成不帶‘.csv’后綴的文件名組成的列表
    final_name = [item +'.csv' for item in name]#生成‘.csv’后綴的文件名組成的列表
    return final_name#輸出由有‘.csv’后綴的文件名組成的列表