python知識總結之csv檔案的寫入與讀取

推薦學習：

CSV (Comma Separated Values)，即逗號分隔值（也稱字元分隔值，因為分隔符可以不是逗號），是一種常用的文字格式，用以儲存表格資料，包括數位或者字元。很多程式在處理資料時都會碰到csv這種格式的檔案。python自帶了csv模組，專門用於處理csv檔案的讀取

csv的寫入

通過建立writer物件，主要用到2個方法。一個是writerow，寫入一行。另一個是writerows寫入多行

使用DictWriter 可以使用字典的方式把資料寫入進去

第一種寫入方法(通過建立writer物件)

先來說一下第一種寫入的方法：通過建立writer物件寫入（每次寫入一行）
步驟：1.建立資料和表頭2.建立writer物件3.寫表頭4.遍歷列表，將每一行資料寫入csv
程式碼如下：

import csv

person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]# 表頭header = ['name', 'age', 'height']with open('person.csv', 'w', encoding='utf-8') as file_obj:
    # 1:建立writer物件
    writer = csv.writer(file_obj)
    # 2:寫表頭
    writer.writerow(header)
    # 3:遍歷列表，將每一行的資料寫入csv
    for p in person:
        writer.writerow(p)

寫入完就會在當前目錄下出現一個person.csv檔案，滑鼠右鍵點選show in Explorer開啟person.csv檢視

開啟以後會發現寫入的資料中間會換行
居然：那麼應該怎麼解決這個問題呢
hacker：很簡單啊
只需要在寫入資料的時候加上一個引數 newline=’'為了防止換行寫入
改正後的程式碼如下：

import csv# 資料person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]# 表頭header = ['name', 'age', 'height']with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 建立物件
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 遍歷，將每一行的資料寫入csv
    for p in person:
        writer.writerow(p)

✅通過建立writer物件（一次性寫入多行）
步驟：1.建立資料和表頭2.建立writer物件3.寫表頭4.在writerows裡傳入你要處理的資料

import csv# 資料person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]# 表頭header = ['name', 'age', 'height']with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 建立物件
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 3.寫入資料(一次性寫入多行)
    writer.writerows(person)

寫入結果如下：

第二種寫入方法(使用DictWriter可以使用字典的方式將資料寫入)

注意事項：使用字典的方式寫入要注意傳遞的資料格式必須是字典
如果不是字典的話會報錯

AttributeError: ‘tuple’ object has no attribute ‘keys’

步驟1.建立資料和表頭( 資料必須是字典格式)2.建立DictWriter物件3.寫表頭4.寫入資料

import csv# 資料person = [
    {'name': 'xxx', 'age': 18, 'height': 193},
    {'name': 'yyy', 'age': 18, 'height': 182},
    {'name': 'zzz', 'age': 19, 'height': 185},]# 表頭header = ['name', 'age', 'height']with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 1.建立DicetWriter物件
    dictWriter = csv.DictWriter(file_obj, header)
    # 2.寫表頭
    dictWriter.writeheader()
    # 3.寫入資料(一次性寫入多行)
    dictWriter.writerows(person)

csv的讀取

通過reader()讀取

import csvwith open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.建立reader物件
    reader = csv.reader(file_obj)
    print(reader)

如果直接列印會返回csv.reader物件，這時需要遍歷列表

<_csv.reader object at 0x000001FB8CE655F8>

改正程式碼如下：

import csvwith open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.建立reader物件
    reader = csv.reader(file_obj)
    # 2.遍歷進行讀取資料
    for r in reader:
        print(r)

讀取結果如下：

['name', 'age', 'height']['xxx', '18', '193']['yyy', '18', '182']['zzz', '19', '185']

如果想列印列表的某一個值，可以使用索引列印

print(r[0])

name
xxx
yyy
zzz

通過dictreader()讀取

import csvwith open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.建立reader物件
    dictReader = csv.DictReader(file_obj)
    # 2.遍歷進行讀取資料
    for r in dictReader:
        print(r)

返回結果如下：

OrderedDict([('name', 'xxx'), ('age', '18'), ('height', '193')])OrderedDict([('name', 'yyy'), ('age', '18'), ('height', '182')])OrderedDict([('name', 'zzz'), ('age', '19'), ('height', '185')])

這時我們如果要取到某一個值就需要指定鍵去尋找值

print(r['name'])

xxx
yyy
zzz

以上就是python基礎教學之csv檔案的寫入和讀取，如果有改進的建議，歡迎在評論區留言奧~

推薦學習：

以上就是python知識總結之csv檔案的寫入與讀取的詳細內容，更多請關注TW511.COM其它相關文章！