Python爬蟲學習筆記（文件存儲：CSV存儲）

CSV：以純文本的形式寫入表格數據

#寫入CSV文件
import csv

with open('data.csv','w',encoding='utf-8') as csvFile:
    writer = csv.writer(csvFile,delimiter=' ') #初始化寫入對象,修改列之間的分隔符爲空格
    writer.writerow(['id','name','age']) #寫入每行數據
    writer.writerow(['10001','Mike','20'])
    writer.writerow(['1002','Bob',22])
    writer.writerow(['10003','Jordan',21])
    #等價於二維列表writer.writerow([['id','name','age'],['10001','Mike','20'],['10003','Jordan',21]])

#用字典來表示
with open('data1.csv','w',encoding='utf-8') as csvFile1:
    fieldnames = ['id','name','age'] #定義字段
    writer1 = csv.DictWriter(csvFile1,fieldnames= fieldnames) #初始化一個字典寫入對象
    writer1.writeheader() #先寫入頭 信息
    writer1.writerow({'id':'10001','name':'Mike','age':20}) #寫入相應的字典
    writer1.writerow({'id':'10002','name':'Bob','age':22})
    writer1.writerow({'id':'10003','name':'Jordan','age':21})

 

#讀CSV文件
with open('data1.csv','r',encoding='utf-8') as csvFile2:
    reader = csv.reader(csvFile2)
    for row in reader:
        print(row)

 

利用read_csv()將數據從CSV中讀取出來

import pandas as pd
df = pd.read_csv('data.csv')
print(df)