python基礎教學之csv檔案的寫入與讀取

2022-08-23 14:01:48

csv的簡單介紹

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

csv的寫入

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

2使用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 csv

with 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 csv

with 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 csv

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

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