快捷導(dǎo)航

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

更新時(shí)間：2022年08月23日 11:08:14 作者：honker707

CSV即逗號(hào)分隔值（也稱字符分隔值,因?yàn)榉指舴梢圆皇嵌禾?hào)）,是一種常用的文本格式,用以存儲(chǔ)表格數(shù)據(jù),包括數(shù)字或者字符,下面這篇文章主要給大家介紹了關(guān)于python基礎(chǔ)教程之csv文件的寫入與讀取的相關(guān)資料,需要的朋友可以參考下

csv的簡(jiǎn)單介紹

CSV (Comma Separated Values)，即逗號(hào)分隔值（也稱字符分隔值，因?yàn)榉指舴梢圆皇嵌禾?hào)），是一種常用的文本格式，用以存儲(chǔ)表格數(shù)據(jù)，包括數(shù)字或者字符。很多程序在處理數(shù)據(jù)時(shí)都會(huì)碰到csv這種格式的文件。python自帶了csv模塊，專門用于處理csv文件的讀取

csv的寫入

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

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

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

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

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

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

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

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

hacker：很簡(jiǎn)單啊

只需要在寫入數(shù)據(jù)的時(shí)候加上一個(gè)參數(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)建對(duì)象
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 遍歷，將每一行的數(shù)據(jù)寫入csv
    for p in person:
        writer.writerow(p)

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

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

寫入結(jié)果如下：

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

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

如果不是字典的話會(huì)報(bào)錯(cuò)

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

步驟1.創(chuàng)建數(shù)據(jù)和表頭( 數(shù)據(jù)必須是字典格式)2.創(chuàng)建DictWriter對(duì)象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對(duì)象
    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對(duì)象
    reader = csv.reader(file_obj)
    print(reader)

如果直接打印會(huì)返回csv.reader對(duì)象，這時(shí)需要遍歷列表

<_csv.reader object at 0x000001FB8CE655F8>

改正代碼如下：

import csv

with open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.創(chuàng)建reader對(duì)象
    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']

如果想打印列表的某一個(gè)值，可以使用索引打印

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對(duì)象
    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')])

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