Python读取文件夹下所有csv文件并将数据按文件名写入字典

Posted 2021-03-01 梦的思维

 

import numpy as np

import os

path = ‘pwd/‘#指定文件所在路径

filetype =‘.csv‘#指定文件类型

def get_filename(path,filetype):

    name =[]

    final_name = []

    a =[]

    data=[]

    di={}

    for root,dirs,files in os.walk(path):

        for i in files:

            if filetype in i:

                a.append(i)

                di[i]=pd.read_csv(path+i)#读取csv文件数据并写入字典

                name.append(i.replace(filetype,‘‘))#生成不带‘.csv’后缀的文件名组成的列表

    final_name = [item +‘.csv‘ for item in name]#生成‘.csv’后缀的文件名组成的列表

    data =[path+ i for i in a]#路径加文件名

    return final_name,data,di

file,path,di=get_filename(path,filetype)

import numpy as npimport ospath = ‘WeatherData/‘#指定文件所在路径filetype =‘.csv‘#指定文件类型def get_filename(path,filetype):    name =[]    final_name = []    a =[]    data=[]    di={}    for root,dirs,files in os.walk(path):        for i in files:            if filetype in i:                a.append(i)                di[i]=pd.read_csv(path+i)                name.append(i.replace(filetype,‘‘))#生成不带‘.csv’后缀的文件名组成的列表    final_name = [item +‘.csv‘ for item in name]#生成‘.csv’后缀的文件名组成的列表    data =[path+ i for i in a]    return final_name,data,di#输出由有‘.csv’后缀的文件名组成的列表file,path,di=get_filename(path,filetype)