今天說一下使用python讀寫csv文件。 讀寫csv文件可以使用基礎python實現,或者使用csv模塊、pandas模塊實現。

基礎python讀寫csv文件

讀寫單個CSV

以下為通過基礎python讀取CSV文件的代碼,請注意,若字段中的值包含有","且該值沒有被引號括起來,則無法通過以下的簡單代碼獲取準確的數據。

inputFile="要讀取的文件名"outputFile=“寫入數據的csv文件名”with open(inputFile,"r")  as fileReader:  with open(outputFile,"w") as fileWriter:    for row in fileReader:      fileWriter.write(row)

讀取多個csv文件并寫入至一個csv文件

讀寫文件的代碼與讀寫單個csv文件大致相同,但需要利用glob模塊以及os模塊獲取需要讀取的文件名。代碼如下:

import osimport globinputPath="讀取csv文件的路徑"outputFile="寫入數據的csv文件名"firstFile=Truefor file in glob.glob(os.path.join(inputPath,"*.csv")):    with open(file,'r') as fileReader:        with open(outputFile,"a") as fileWriter:            if firstFile:                for row in fileReader:                    fileWriter.write(row)                firstFile=False                fileWriter.write("n")            else:                header=fileReader.readline()                for row in fileReader:                    fileWriter.write(row)

通過pandas模塊讀寫csv文件

讀寫單個CSV

pandas的dataframe類型有相應的方法能讀取csv文件,代碼如下:

import pandas as pdinputFile="要讀取的文件名"outputFile=“寫入數據的csv文件名”df=pd.read_csv(inputFile)df.to_csv(outputFile)

請注意,若字段中的值包含有","且該值沒有被引號括起來,則無法通過以下的簡單代碼獲取準確的數據。

讀取多個csv文件并寫入至一個csv文件

import osimport globimport pandas as pdinputPath="讀取csv文件的路徑"outputFile="寫入數據的csv文件名"dataFrameList=[]for file in glob.glob(os.path.join(inputPath,"*.csv")):     df=pd.read_csv(file)    dataFrameList.append(df)allDataFrame=pd.concat(dataFrameList,axis=0,ignore_index=True)allDataFrame.to_csv(outputFile)

通過csv模塊讀寫csv文件

讀寫單個CSV文件

代碼如下:

import csvinputFile="要讀取的文件名"outputFile=“寫入數據的csv文件名”with open(inputFile,"r",newline='') as fileReader:    with open(outputFile,"w",newline='') as fileWriter:        csvReader=csv.reader(fileReader,delimiter=',')        csvWriter=csv.writer(fileWriter,delimiter=',')        for row in csvReader:            print(row)            csvWriter.writerow(row)

讀取多個csv文件并寫入至一個csv文件

思路與上述用基礎python讀取多個csv文件大體相同,代碼如下:

import csvimport globimport osinputPath=r"讀取csv文件的路徑"outputFile=r"輸出文件的路徑"firstFile=Truefor file in glob.glob(os.path.join(inputFile,"*.csv")):    with open(file,"r") as fileReader:        with open(outputFile,"a") as fileWriter:            csvReader=csv.reader(fileReader)            csvWriter=csv.writer(fileWriter)            if firstFile:                for row in csvReader:                    csvWriter(row)                firstFile=False            else:                header=next(csvReader,None)                for row in csvReader:                    csvWriter(row)