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)