站長(zhǎng)資訊網(wǎng)本篇文章給大家?guī)砹岁P(guān)于python的相關(guān)知識(shí)，其中主要介紹了csv文件的寫入與讀取的相關(guān)問題，CSV是一種常用的文本格式，用以存儲(chǔ)表格數(shù)據(jù)，包括數(shù)字或者字符，希望對(duì)大家有幫助。
推薦學(xué)習(xí)：python教程
CSV (Comma Separated Values)，即逗號(hào)分隔值（也稱字符分隔值，因?yàn)榉指舴梢圆皇嵌禾?hào)），是一種常用的文本格式，用以存儲(chǔ)表格數(shù)據(jù)，包括數(shù)字或者字符。很多程序在處理數(shù)據(jù)時(shí)都會(huì)碰到csv這種格式的文件。python自帶了csv模塊，專門用于處理csv文件的讀取
csv的寫入
通過創(chuàng)建writer對(duì)象，主要用到2個(gè)方法。一個(gè)是writerow，寫入一行。另一個(gè)是writerows寫入多行
使用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 csvwith 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 csvwith 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 csvwith 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è)值就需要指定鍵去尋找值
print(r['name'])
xxx yyy zzz
