python 读取指定文件夹中的指定文件类型的文件名

Z时代
2024-01-10
分类：综合

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