快捷導(dǎo)航

python讀取當(dāng)前目錄下的CSV文件數(shù)據(jù)

更新時(shí)間：2020年03月11日 08:41:58 作者：鄭旭濤

這篇文章主要為大家詳細(xì)介紹了python讀取當(dāng)前目錄下的CSV文件數(shù)據(jù)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

在處理數(shù)據(jù)的時(shí)候，經(jīng)常會(huì)碰到CSV類型的文件，下面將介紹如何讀取當(dāng)前目錄下的CSV文件，步驟如下

1、獲取當(dāng)前目錄所有的CSV文件名稱：

#創(chuàng)建一個(gè)空列表，存儲(chǔ)當(dāng)前目錄下的CSV文件全稱
file_name = []
 
#獲取當(dāng)前目錄下的CSV文件名
def name():
  #將當(dāng)前目錄下的所有文件名稱讀取進(jìn)來(lái)
  a = os.listdir()
  for j in a:
    #判斷是否為CSV文件，如果是則存儲(chǔ)到列表中
    if os.path.splitext(j)[1] == '.csv':
      file_name.append(j)

2、將CSV文件讀取進(jìn)來(lái)：

#將CSV文件內(nèi)容導(dǎo)入到csv_storage列表中
def csv_new(storage,name):
  #創(chuàng)建一個(gè)空列表，用于存儲(chǔ)CSV文件數(shù)據(jù)
  csv_storage = []
  with codecs.open(storage, 'r', encoding='utf-8') as fp:
    fp_key = csv.reader(fp)
    for csv_key in fp_key:
      csv_reader = csv.DictReader(fp, fieldnames=csv_key)
      for row in csv_reader:
        csv_dict = dict(row)
        csv_storage.append(csv_dict)

3、連續(xù)讀取多個(gè)CSV文件：

設(shè)置一個(gè)for循環(huán)，將第一部分讀取到的文件名稱逐個(gè)傳遞給讀取文件的函數(shù)，全部代碼如下所示：

import codecs
import csv
import os
 
#創(chuàng)建一個(gè)空列表，存儲(chǔ)當(dāng)前目錄下的CSV文件全稱
file_name = []
 
#獲取當(dāng)前目錄下的CSV文件名
def name():
  #將當(dāng)前目錄下的所有文件名稱讀取進(jìn)來(lái)
  a = os.listdir()
  for j in a:
    #判斷是否為CSV文件，如果是則存儲(chǔ)到列表中
    if os.path.splitext(j)[1] == '.csv':
      file_name.append(j)
 
 
#將CSV文件內(nèi)容導(dǎo)入到csv_storage列表中
def csv_new(storage):
  #創(chuàng)建一個(gè)空列表，用于存儲(chǔ)CSV文件數(shù)據(jù)
  csv_storage = []
  with codecs.open(storage, 'r', encoding='utf-8') as fp:
    fp_key = csv.reader(fp)
    for csv_key in fp_key:
      csv_reader = csv.DictReader(fp, fieldnames=csv_key)
      for row in csv_reader:
        csv_dict = dict(row)
        csv_storage.append(csv_dict)
  for i in csv_storage:
    print(i)
 
#主要運(yùn)行函數(shù)
if __name__ == '__main__':
  #運(yùn)行獲取當(dāng)前目錄下所有的CSV文件
  name()
  #將多個(gè)CSV文件逐個(gè)讀取
  for name in file_name:
    csv_new(name)
  print(file_name)

4、最終的結(jié)果輸出：