快捷導(dǎo)航

Python數(shù)據(jù)讀寫之Python讀寫CSV文件

更新時(shí)間：2022年06月28日 14:05:53 作者：還是要努力呀！

這篇文章主要介紹了Python數(shù)據(jù)讀寫之Python讀寫CSV文件，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，感興趣的小伙伴可以參考一下

1. 讀取CSV文件 csv.reader()

該方法的作用相當(dāng)于就是通過 ',' 分割csv格式的數(shù)據(jù)，并將分割好的每行數(shù)據(jù)存入列表中，并且還去除了每行最后分割產(chǎn)生的數(shù)據(jù)尾部的空格、換行符、制表符等等。

import csv
with open('data.csv',mode='r',encoding='utf-8-sig',newline='') as File:
    # 使用csv.reader()將文件中的每行數(shù)據(jù)讀入到一個(gè)列表中
    reader = csv.reader(File, delimiter=',', quotechar=',', quoting=csv.QUOTE_MINIMAL)
    # 遍歷列表將數(shù)據(jù)按行輸出
    for row in reader:
        print(row)

直接通過索引來訪問每行數(shù)據(jù)的列元素

for row in reader:
    print(row[0])

csv.DictReader()

該方法將文件中的每行數(shù)據(jù)保存在 OrderedDict 中，此數(shù)據(jù)類型類似于嵌套著元組的列表，每個(gè)元組中的第一個(gè)元素為鍵，第二個(gè)元素為值，其中元組中的鍵來自于CSV數(shù)據(jù)中的表頭信息。

import csv
results = []
with open('data.csv',mode='r',encoding='utf-8-sig',newline='') as File:
    reader = csv.DictReader(File)
    for row in reader:
        print(row)

通過鍵值來讀取部分信息

 for row in reader:
        print(row['MakeSpan'],row['WaitTime'])

關(guān)于DictReader()的更多內(nèi)容尚未學(xué)習(xí)整理，如有需要可參考其他文章：CSV.DictReader()方法

2. 寫入CSV文件

首先需要導(dǎo)入讀寫csv需要用的包：

import csv

使用open()函數(shù)打開一個(gè)文件，open()函數(shù)常用的參數(shù)：

file：文件路徑、文件名
mode：打開模式，r(只讀)，w(只寫)，a(追加模式)
newline：每行之間是否存在空行，默認(rèn)存在空行，' '：沒有空行。

myFile = open('example2.csv', 'w', newline='')

csv.writer模塊，用于將數(shù)據(jù)寫入CSV：

csvfile：這能夠是帶有write()方法的任何對(duì)象。
dialect=‘excel’：一個(gè)可選參數(shù)，用于定義特定于特定CSV的一組參數(shù)。
fmtparam：可選參數(shù)，可用于覆蓋現(xiàn)有的格式設(shè)置參數(shù)。

writer = csv.writer(myFile)

使用writerow()和writerows()將數(shù)據(jù)寫入到CSV文件中：

writerow()：將數(shù)據(jù)存到csv文件中的一行里，每個(gè)元素占用一個(gè)單元格
writerows()：將數(shù)據(jù)中的每一個(gè)列表存到CSV文件中的一行里，列表中每個(gè)元素占用一個(gè)單元格

myData1 = [["這", "是", "writerow", "的", "效", "果"],
           ["這", "是", "writerow", "的", "效", "果"],
           ["這", "是", "writerow", "的", "效", "果"]]
           
myData2 = [["這", "是", "writerows", "的", "效", "果"],
           ["這", "是", "writerows", "的", "效", "果"],
           ["這", "是", "writerows", "的", "效", "果"]]
myFile = open('example2.csv', 'w', newline='')
with myFile:
    writer = csv.writer(myFile)
    writer.writerow(myData1) 
    writer.writerows(myData2)

寫入文件后的效果如下所示：