Python讀取文件夾下所有csv文件並將數據按文件名寫入字典

 

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)