站長資訊網本篇文章給大家帶來了關于python的相關知識，其中主要介紹了csv文件的寫入與讀取的相關問題，CSV是一種常用的文本格式，用以存儲表格數據，包括數字或者字符，希望對大家有幫助。
推薦學習：python教程
CSV (Comma Separated Values)，即逗號分隔值（也稱字符分隔值，因為分隔符可以不是逗號），是一種常用的文本格式，用以存儲表格數據，包括數字或者字符。很多程序在處理數據時都會碰到csv這種格式的文件。python自帶了csv模塊，專門用于處理csv文件的讀取
csv的寫入
通過創建writer對象，主要用到2個方法。一個是writerow，寫入一行。另一個是writerows寫入多行
使用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 csvwith 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 csvwith 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 csvwith 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')])
這時我們如果要取到某一個值就需要指定鍵去尋找值
print(r['name'])
xxx yyy zzz
