? ?CSV马靠,全稱為Comma-Separated Values卷拘,中文可以叫作逗號分隔值或字符分隔值仿耽,其文件以純文本形式存儲表格數(shù)據眯搭。它比Excel文件更加簡介窥翩,XLS文本是電子表格,它包含了文本鳞仙、數(shù)值、公式和格式等內容笔时,而CSV中不包含這些內容棍好,就是特定字符分隔的純文本,結構簡單清晰允耿。所以借笙,有時候用CSV來保存數(shù)據是比較方便的
?本文用python代碼實現(xiàn)CSV文件的寫入和讀取方式。這也是一種常用的數(shù)據存儲方式较锡,需要熟練掌握
1.寫入
?首先业稼,打開data.csv文件,然后指定打開的模式為w(即寫入)蚂蕴,獲得文件句柄低散,隨后調用csv庫的writer()方法初始化寫入對象俯邓,傳入該句柄,然后調用writerow()方法傳入每行的數(shù)據即可完成寫入熔号。
import csv
with open('data.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['id', 'name', 'age'])
writer.writerow(['10001', 'Mike', 20])
writer.writerow(['10002', 'Bob', 22])
writer.writerow(['10003', 'Jordan', 21])
?運行結束后稽鞭,會生成一個名為data.csv的文件,此時數(shù)據就成功寫入了引镊。直接以文本形式打開的話朦蕴,其內容如下
?可以看到,寫入的文本默認以逗號分隔弟头,調用一次writerow()方法即可寫入一行數(shù)據吩抓。用Excel打開的結果如下
?一般情況下,爬蟲爬取的都是結構化數(shù)據赴恨,我們一般會用字典來表示琴拧。在csv庫中也提供了字典的寫入方式,示例如下:
import csv
with open('data.csv', 'w') as csvfile:
fieldnames = ['id', 'name', 'age']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'id': '10001', 'name': 'Mike', 'age': 20})
writer.writerow({'id': '10002', 'name': 'Bob', 'age': 22})
writer.writerow({'id': '10003', 'name': 'Jordan', 'age': 21})
?這里先定義3個字段嘱支,用fieldnames表示蚓胸,然后將其傳給DictWriter來初始化一個字典寫入對象,接著可以調用writeheader()方法先寫入頭信息除师,然后再調用writerow()方法傳入相應字典即可沛膳。最終寫入的結果是完全相同的
另外,如果想追加寫入的話汛聚,可以修改文件的打開模式锹安,即將open()函數(shù)的第二個參數(shù)改成a,代碼如下:
import csv
with open('data.csv', 'a') as csvfile:
fieldnames = ['id', 'name', 'age']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writerow({'id': '10004', 'name': 'Durant', 'age': 22})
這樣在上面的基礎上再執(zhí)行這段代碼倚舀,新行會添加成功,如下:
?如果要寫入中文內容的話叹哭,可能會遇到字符編碼的問題,此時需要給open()參數(shù)指定編碼格式痕貌。比如风罩,這里再寫入一行包含中文的數(shù)據,代碼需要改寫如下:
import csv
with open('data.csv', 'a', encoding='utf-8', newline='') as csvfile:
fieldnames = ['id', 'name', 'age']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writerow({'id': '10005', 'name': '王偉', 'age': 22})
運行結果如下:
2.讀取
我們同使用csv庫來讀取CSV文件舵稠。例如超升,將剛才寫入的文件內容讀取出來,相關代碼如下
import csv
with open('data.csv','r', encoding='utf-8') as csvfile:
# 調用csv模塊的reader函數(shù)讀文件
lines = csv.reader(csvfile)
for line in lines:
print(line)
運行結果如下:
另外哺徊,如果接觸過pandas的話室琢,可以利用read_csv()方法將數(shù)據從CSV中讀取出來,例如
import pandas as pd
df = pd.read_csv('data.csv')
print(df)
運行結果如下:
在做數(shù)據分析的時候落追,csv儲存方法用得比較多盈滴,也是一種比較方便地讀取CSV文件的方法。