怎么使用Python讀寫CSV文件

這篇文章主要介紹了怎么使用Python讀寫CSV文件的相關知識，內容詳細易懂，操作簡單快捷，具有一定借鑒價值，相信大家閱讀完這篇怎么使用Python讀寫CSV文件文章都會有所收獲，下面我們一起來看看吧。

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

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

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

直接通過 索引 來訪問每行數據的列元素

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

csv.DictReader()

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

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'])

2. 寫入CSV文件

首先需要導入讀寫csv需要用的包：

import csv

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

• file：文件路徑、文件名

• mode：打開模式，r(只讀)，w(只寫)，a(追加模式)

• newline：每行之間是否存在空行，默認存在空行，' '：沒有空行。

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

csv.writer模塊，用于將數據寫入CSV：

• csvfile：這 能夠 是帶有write()方法的任何對象。

• dialect=‘excel’：一個可選參數，用于定義特定于特定CSV的一組參數。

• fmtparam：可選參數，可用于覆蓋現有的格式設置參數。

writer = csv.writer(myFile)

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

• writerow()：將數據存到csv文件中的一行里，每個元素占用一個單元格

• writerows()：將數據中的每一個列表存到CSV文件中的一行里，列表中每個元素占用一個單元格

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)

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

關于“怎么使用Python讀寫CSV文件”這篇文章的內容就介紹到這里，感謝各位的閱讀！相信大家對“怎么使用Python讀寫CSV文件”知識都有一定的了解，大家如果還想學習更多知識，歡迎關注億速云行業資訊頻道。