快捷導(dǎo)航

python基礎(chǔ)教程之csv格式文件的寫入與讀取

更新時間：2022年03月16日 13:00:22 作者：hacker707

逗號分隔值（Comma-Separated Values,CSV,也稱為字符分隔值,分隔字符也可以不是逗號）,新這篇文章主要給大家介紹了關(guān)于python基礎(chǔ)教程之csv格式文件的寫入與讀取的相關(guān)資料,需要的朋友可以參考下

csv的簡單介紹

CSV (Comma Separated Values)，即逗號分隔值（也稱字符分隔值，因為分隔符可以不是逗號），是一種常用的文本格式，用以存儲表格數(shù)據(jù)，包括數(shù)字或者字符。很多程序在處理數(shù)據(jù)時都會碰到csv這種格式的文件。python自帶了csv模塊，專門用于處理csv文件的讀取

csv的寫入

1通過創(chuàng)建writer對象，主要用到2個方法。一個是writerow，寫入一行。另一個是writerows寫入多行

2使用DictWriter 可以使用字典的方式把數(shù)據(jù)寫入進(jìn)去

第一種寫入方法(通過創(chuàng)建writer對象)

?先來說一下第一種寫入的方法：通過創(chuàng)建writer對象寫入（每次寫入一行）

步驟：1.創(chuàng)建數(shù)據(jù)和表頭2.創(chuàng)建writer對象3.寫表頭4.遍歷列表，將每一行數(shù)據(jù)寫入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:創(chuàng)建writer對象
    writer = csv.writer(file_obj)
    # 2:寫表頭
    writer.writerow(header)
    # 3:遍歷列表，將每一行的數(shù)據(jù)寫入csv
    for p in person:
        writer.writerow(p)

寫入完就會在當(dāng)前目錄下出現(xiàn)一個person.csv文件，鼠標(biāo)右鍵點(diǎn)擊show in Explorer打開person.csv查看

打開以后會發(fā)現(xiàn)寫入的數(shù)據(jù)中間會換行

居然：那么應(yīng)該怎么解決這個問題呢

hacker：很簡單啊

只需要在寫入數(shù)據(jù)的時候加上一個參數(shù) newline=’'為了防止換行寫入

改正后的代碼如下：

import csv

# 數(shù)據(jù)
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:
    # 創(chuàng)建對象
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 遍歷，將每一行的數(shù)據(jù)寫入csv
    for p in person:
        writer.writerow(p)

?通過創(chuàng)建writer對象（一次性寫入多行）

步驟：1.創(chuàng)建數(shù)據(jù)和表頭2.創(chuàng)建writer對象3.寫表頭4.在writerows里傳入你要處理的數(shù)據(jù)

import csv

# 數(shù)據(jù)
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:
    # 創(chuàng)建對象
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 3.寫入數(shù)據(jù)(一次性寫入多行)
    writer.writerows(person)

寫入結(jié)果如下：

第二種寫入方法(使用DictWriter可以使用字典的方式將數(shù)據(jù)寫入)

注意事項：使用字典的方式寫入要注意傳遞的數(shù)據(jù)格式必須是字典

如果不是字典的話會報錯

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

步驟1.創(chuàng)建數(shù)據(jù)和表頭( 數(shù)據(jù)必須是字典格式)2.創(chuàng)建DictWriter對象3.寫表頭4.寫入數(shù)據(jù)

import csv

# 數(shù)據(jù)
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.創(chuàng)建DicetWriter對象
    dictWriter = csv.DictWriter(file_obj, header)
    # 2.寫表頭
    dictWriter.writeheader()
    # 3.寫入數(shù)據(jù)(一次性寫入多行)
    dictWriter.writerows(person)

csv的讀取

通過reader()讀取

import csv

with open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.創(chuàng)建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.創(chuàng)建reader對象
    reader = csv.reader(file_obj)
    # 2.遍歷進(jìn)行讀取數(shù)據(jù)
    for r in reader:
        print(r)

讀取結(jié)果如下：

['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.創(chuàng)建reader對象
    dictReader = csv.DictReader(file_obj)
    # 2.遍歷進(jìn)行讀取數(shù)據(jù)
    for r in dictReader:
        print(r)

返回結(jié)果如下：

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

這時我們?nèi)绻〉侥骋粋€值就需要指定鍵去尋找值