csv模块

一、csv的写入

csv是python的内置库，我们可以利用这个数据库将数据写入表格

csv写入，有两种方式

1.csv普通序列的写入与保存

import csv
data = [('肖申克的救赎', '9.7', '希望让人自由'), ('霸王别姬', '9.6', '风华绝代'), ('阿甘正传', '9.5', '一部美国近现代史')]

# 写入，表头   以元组形式写入
head = ('title','star','quote')

# 保存数据
with open ('douban.csv','w',encoding='utf-8-sig',newline='') as f:
    # 首先是创建csv对象
    writer = csv.writer(f)    #csv.writer()中可以传入一个文件对象
    #写入表头
    writer.writerow(head)
    #写入内容
    writer.writerows(data)     #多行写入
#会换行，加入newline=''
    for d in data:           #遍历列表
        writer.writerow(d)     #一行一行写入

思路：

1.csv内置模块，直接导入就好

2.数据都是需要爬取下来，数据处理好格式

3.要有表头

4.保存数据，文件名后缀.csv，encoding='utf-8-sig',newline=' ',其中newline目的是不换行

5.创建csv对象，writer=csv.writer( )

6.写入表头 writer.writerow( )

7.写入内容 writer.writerows( )

2.csv字典数据的写入与保存

import csv
data = [
    {'标题': '肖申克的救赎', '评分': '9.7', '引言': '希望让人自由'},
    {'标题': '霸王别姬', '评分': '9.6', '引言': '风华绝代'},
    {'标题': '阿甘正传', '评分': '9.5', '引言': '一部美国近现代史'}
]
#设置标题
head = ('标题','评分','引言')
#保存文件
with open('dou.csv','w',encoding='utf-8-sig',newline='') as f:
    #创建csv对象
    writer = csv.DictWriter(f , fieldnames=head)     #写入两个参数
    #Ewriter.writerow(head)   写入表头
    writer.writeheader()    #调用
    writer.writerows(data)

思路：

1.导入csv模块

2.准备数据，数据以列表嵌套字典的格式

3.设置表头，跟字典中的key一一对应

4.DictWriter(f , fieldnames=head) 必须有两个参数，第一个文件对象，第二个参数表头

5.写入表头：writeheader()

6.写入对应的数据

二、csv的读取

1.普通序列的读取

import csv
#第一种方式的读取
with open('douban.csv','r',encoding='utf-8-sig') as f:     #../返回上一级
    #创建读取对象
    reader = csv.reader(f)
    print(reader)     # <_csv.reader object at 0x000002A8CEC8D198> 生成器对象
    for i in reader:
        print(i)    #列表

2.字典序列的读取

#第二种方式的读取  字典
import csv
with open('douban.csv','r',encoding='utf-8-sig') as f:
    #创建读取对象
    reder=csv.DictReader(f)
    print(reader)      #生成器对象
    for i in reder:
        print(i)    #OrderedDict([('title', '肖申克的救赎'), ('star', '9.7'), ('quote', '希望让人自由')])
        print(i['title'])   #以字典形式取