本篇文章給大家?guī)砹岁P(guān)于python的相關(guān)知識,其中主要介紹了csv文件的寫入與讀取的相關(guān)問題,CSV是一種常用的文本格式,用以存儲表格數(shù)據(jù),包括數(shù)字或者字符,希望對大家有幫助。
推薦學(xué)習(xí):python教程
CSV (Comma Separated Values),即逗號分隔值(也稱字符分隔值,因為分隔符可以不是逗號),是一種常用的文本格式,用以存儲表格數(shù)據(jù),包括數(shù)字或者字符。很多程序在處理數(shù)據(jù)時都會碰到csv這種格式的文件。python自帶了csv模塊,專門用于處理csv文件的讀取
csv的寫入
通過創(chuàng)建writer對象,主要用到2個方法。一個是writerow,寫入一行。另一個是writerows寫入多行
使用DictWriter 可以使用字典的方式把數(shù)據(jù)寫入進去
第一種寫入方法(通過創(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)
寫入完就會在當前目錄下出現(xiàn)一個person.csv文件,鼠標右鍵點擊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 csvwith 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 csvwith open('person.csv', 'r', encoding='utf-8') as file_obj: # 1.創(chuàng)建reader對象 reader = csv.reader(file_obj) # 2.遍歷進行讀取數(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 csvwith open('person.csv', 'r', encoding='utf-8') as file_obj: # 1.創(chuàng)建reader對象 dictReader = csv.DictReader(file_obj) # 2.遍歷進行讀取數(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)绻〉侥骋粋€值就需要指定鍵去尋找值
print(r['name'])
xxx yyy zzz