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’后缀的文件名组成的列表

 

转载于:https://www.cnblogs.com/qiu-hua/p/7810764.html