0
点赞
收藏
分享

微信扫一扫

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)


为啥写这篇文章?

前面几篇文章我们介绍了正则表达式的使用,主要还是介绍数据解析方面的知识点。这篇文章开始我们就将介绍数据存储方面的知识点了。
前面几篇文章还没看的小伙伴,可以抓紧看看啦!!!!
​用正则表达式爬取古诗文网站,边玩边学【python爬虫入门进阶】(09) 本文主要介绍csv文件的读写操作,文件简单易懂。

CSV文件是什么?

CSV即Comma Separate Values,这种文件格式经常用来作为不同程序之间的数据交互的格式。
具体文件格式

1.每条记录占一行 以逗号为分隔符
2.逗号前后的空格会被忽略
3.字段中包含有逗号,该字段必须用双引号括起来
4.字段中包含有换行符,该字段必须用双引号括起来
5.字段前后包含有空格,该字段必须用双引号括起来
6.字段中的双引号用两个双引号表示
7.字段中如果有双引号,该字段必须用双引号括起来
8.第一条记录,可以是字段名

如何读取CSV文件

通过reader方法读取

首先,以content_test.csv文件为例进行说明。

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)_python

  1. 读取csv中的所有列

import csv
with open('content_test.csv', 'r') as fp:
reader = csv.reader(fp)
# 过滤掉标题行
next(reader)
for x in reader:
title = x[1]
print(x)
print(title)

这里通过​​csv.reader(fp)​​​ 创建csv_reader对象,通过调用​​next(reader)​​ 方法可以过滤掉标题行。

遍历reader可以获取到每一行的数据,指定 ​​x[1]​​ 可以获取到每行第二个单元格的数据。

最终的运行结果是:

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)_数据_02

通过DictReader方法读取

用DictReader方法创建redader对象, 不会包含标题那行的数据。
reader是一个迭代器。遍历这个迭代器,返回的是一个字典。

import csv
with open('content_test.csv', 'r') as fp:
reader = csv.DictReader(fp)
for x in reader:
value = {'序号': x['序号'], '标题': x['标题']}
print(value)

运行结果是:

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)_爬虫_03

如何向CSV文件中写入数据

与读取csv的方法类似的,向CSV文件中写数据的方法就是通过writer对象来操作。

通过writer方法

import csv
headers = ['用户名', '年龄', '身高']
values = [
('张三', 18, 179),
('李四', 20, 180),
('王五', 25, 177),
('码农飞哥', 18, 171)
]

with open('user_info.csv', 'w', encoding='utf-8', newline='') as fp:
writer = csv.writer(fp)
writer.writerow(headers)
writer.writerows(values)

通过headers来指定CSV的表头。通过values指定单元格的数据。

通过写​​open('user_info.csv', 'w', encoding='utf-8', newline='')​​​文件的mode是w。通过​​encoding='utf-8'​​​指定编码格式为utf-8,如果不指定的话可能会出现中文乱码的情况。指定​​newline=''​​ 指定行与行之间的空格,如果不指定的话则每行之间有空格。

通过​​csv.writer(fp)​​ 创建一个writer对象。

通过writerow方法写入表头

通过writerows方法写入每行数据。

运行结果是:

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)_爬虫_04

通过DictWriter方法

import csv
headers = ['用户名', '年龄', '身高']
values = [
{'用户名': '张三', '年龄': 18, '身高': 179},
{'用户名': '李四', '年龄': 20, '身高': 180},
{'用户名': '王五', '年龄': 25, '身高': 177},
{'用户名': '码农飞哥', '年龄': 18, '身高': 171}
]
with open('user_info2.csv', 'w', encoding='utf-8', newline='') as fp:
writer = csv.DictWriter(fp, headers)
# 写入表头数据的时候需要调用writeheader
writer.writeheader()
writer.writerows(values)

DictWriter的方法在写入时是以字典的形式写入数据。

通过​​csv.DictWriter(fp, headers)​​创建writer对象。

通过 ​​writeheader()​​方法写入表头

通过​​writerows(values)​​ 方法写入表单元格的数据。

CSV文件操作起来还挺方便的【python爬虫入门进阶】(10)_java_05

总结

CSV文件操作起来还挺方便的

粉丝专属福利

软考资料:​​实用软考资料​​

面试题:​​5G 的Java高频面试题​​

学习资料:​​50G的各类学习资料​​

并发编程:回复【并发编程】


举报

相关推荐

0 条评论