大数据时代,各行各业对数据采集的需求日益增多,网络爬虫的运用也更为广泛,越来越多的人开始学习网络爬虫这项技术,K哥爬虫此前已经推出不少爬虫进阶、逆向相关文章,为实现从易到难全方位覆盖,特设【0基础学爬虫】专栏,帮助小白快速入门爬虫,本期为文件存储。
前几期文章中我们已经了解到了请求库、解析库的使用,已经学会了如何向目标网站发送请求以及解析响应信息,那么我们还需要知道如何将数据进行存储。数据存储有很多方式,本期会介绍如何将数据存入文本文件,如 TXT、JSON、CSV 等。
TXT文件存储是最基本的数据存储方式,TXT 文本文件基本上兼容任何平台,因此它的应用也非常广泛。使用代码操作 TXT 文件十分简单,但是缺点也非常明显,不利于对数据进行检索。
Python 中操作 TXT 文件需要使用到内置的 open() 函数。
open(file, mode='r', buffering=None, encoding=None, errors=None, newline=None, closefd=True)
open() 函数有七个参数分别为:
file:文件路径
mode:打开方式,默认为(r)只读模式
buffering:文件缓冲
encoding:字符编码
errors:报错级别
newline:换行模式,默认以\n换行
closefd:控制在关闭文件时是否彻底关闭文件
文件打开方式分为以下几种:
r:以只读方式打开文件
w:以写入方式打开文件。如果文件存在则覆盖原文件,不存在则新建。
x:创建新文件并以写入方式打开文件
a:以追加方式打开文件。如果文件存在,则在结尾位置写入,不存在则新建。
b:二进制模式
t:文本模式
+:读取与写入
wb:以二进制写入方式打开文件。如果文件存在则覆盖原文件,不存在则新建。
w+:以读写方式打开文件。如果文件存在则覆盖原文件,不存在则新建。
wb+:以二进制读写方式打开文件。如果文件存在则覆盖原文件,不存在则新建。
ab:以二进制追加方式打开文件。如果文件存在,则在结尾位置写入,不存在则新建。
ab+:以二进制追加读写方式打开文件。如果文件存在,则在结尾位置写入,不存在则新建。
f = open('data.txt','r',encoding="utf-8") f.seek(2) #从指定下标位置开始读取 data = f.read() f.close()
f = open('data.txt','w',encoding="utf-8") f.write('hello') f.write(' ') f.write('world') f.close() #hello world
上文中可以看到,对一个文件进行读写操作后需要使用 close() 方法关闭文件。如果不使用 close() 关闭文件会导致资源泄露问题。为了避免这个问题和优化代码,我们可以使用上下文管理器来操作文件。
with open('data.txt','w',encoding='uutf-8')as f: f.write('hello') f.write(' ') f.write('world')
当 with 语句块结束后文件会自动关闭,这样不需要再调用 close() 方法。
JSON 全称为 JavaScript Object Notation,也就是 JavaScript 对象标记,是一种轻量级的数据交换格式。人们可以很容易的对 JSON 格式的数据进行阅读和编写,同时也方便了机器进行解析和生成。JSON 采用完全独立于程序语言的文本格式,但是也使用了类 C 语言的习惯(包括C, C++, C#, Java, JavaScript, Perl, Python等)。这些特性使 JSON 成为理想的数据交换语言。
JSON 通过对象和数组的组合来表示数据,而在 JavaScript 中,一切皆为对象,因此任何数据类型都可以通过 JSON 来表示。
在 JavaScript 中 JSON 通常以对象或数组的形式存在。对象形式为被花括号包裹的键值对结构,如 { key1 : value1, key2 : value2 }
,key表示对象的属性,value表示对象的值。数组形式与对象形式的区别就在于它是被方括号所包裹,如[{ key1 : value1}, {key2 : value2}]
。对象形式与数组形式可以自由组合,任意嵌套。JSON中字符串类型的数据需要以双引号包围。
JSON 的这种结构非常简洁清晰,且结构化程度很高,因而被广泛使用。
JSON 是一个数据格式,本质上是纯字符串类型,在 Python 中它可以被转换为对应的数据类型,如对象形式可以被转换为字典,数组类型可以转换为列表。Python 中内置了一个强大的 JSON 库,通过它可以简便的实现 JSON 文件的读写操作。
JSON 库提高了四个方法:
json.dump():将 JSON 对象写入文件
json.dumps():将Python对象编码成 JSON 字符串
json.load():从 JSON 文件中读取数据
json.loads():将已编码的 JSON 字符串解码为Python对象
下面我们以这段数据作为示例:
{ "1001": { "name": "张三", "age": 21, "gender": "female" }, "1002": { "name": "李四", "age": 22, "gender": "male" } }
data.json 文件内容为示例 JSON 数据。
with open('data.json','r',encoding='utf-8')as f: data = f.read() print(type(data)) #输出:<class 'str'>
可以看到,我们直接从 JSON 文件里面读取的数据是字符串类型,这不利于我们对它进行操作,我们可以使用 json.loads() 方法将 JSON 字符串转换为字典类型。
import json with open('data.json','r',encoding='utf-8')as f: data = f.read() print(type(data)) data = json.loads(data) print(type(data)) #输出: #<class 'str'> #<class 'dict'>
我们也可以使用更加简洁的写法,通过 json.load() ,直接对文件操作对象进行解析。
import json data = json.load(open('data.json',encoding='utf-8')) print(type(data)) #输出:<class 'dict'>
使用 write() 方法写入文件时,它只能传入字符串类型的数据。因此我们需要将字典数据先转换为 JSON 字符串,可以通过 json.dumps() 方法实现。
import json data = { "1001": { "name": "张三", "age": 21, "gender": "female" }, "1002": { "name": "李四", "age": 22, "gender": "male" } } print(data) data = json.dumps(data) print(type(data)) with open('data.json','w',encoding="utf-8")as f: f.write(data) #输出: #<class 'dict'> #<class 'str'>
当然,将数据写入文件也有更加简便的方法。
import json data = { "1001": { "name": "张三", "age": 21, "gender": "female" }, "1002": { "name": "李四", "age": 22, "gender": "male" } } json.dump(data,open('data.json','w',encoding='utf-8'),indent=2,ensure_ascii=False)
CSV 全称 Comma-Sparated Values(逗号分隔值),它以纯文本形式存储数据表格。CSV 文件是一个字符序列,它由任意数目的记录组成,记录之间以某种换行符分隔。文件的每条记录由字段和字段值组成,字段间的分隔符是其它字符或字符串,最常见的是逗号或制表符。CSV 并不是一种单一的、明确定义的文本格式,它泛指具备以下特征的任何文件:
纯文本呢,使用某个字符集
由记录组成
每条记录被分隔符分隔为字段
每条记录都有相同的字段序列
Python 中提供了一个内置的 csv 库来对 CSV 文件进行操作。
import csv #writerow默认换行,因此需要通过 newline 将换行符设置为空,避免重复换行 with open('data.csv','w',encoding='utf-8',newline="") as csvfile: #delimiter:指定分隔符,默认为逗号 writer = csv.writer(csvfile,delimiter = " ") #写入一行 writer.writerow(['id', 'name', 'age']) writer.writerow(['1001', 'jack', 15]) writer.writerow(['1002', 'marin', 22]) writer.writerow(['1003', 'alex', 19]) #写入多行 writer.writerows([['1004', 'lucy', 23],['1005', 'john', 30]])
csv库也提供了其它的写入方法
import csv with open('data.csv','w',encoding='utf-8',newline="") as csvfile: #字典写入对象 writer = csv.DictWriter(csvfile,fieldnames=['id','name','age']) writer.writeheader() writer.writerow({'id':'1001','name':'jack','age':21}) writer.writerow({'id': '1002', 'name': 'marin', 'age': 22}) writer.writerows([{'id': '1003', 'name': 'lucy', 'age': 23},{'id': '1004', 'name': 'mike', 'age': 22}])
import csv with open('data.csv','r',encoding='utf-8')as csvfile: reader = csv.reader(csvfile) for row in reader: print(row)
上文中讲到了 TXT、JSON、CSV 三种文件的存储方式,介绍了如何对这些文件进行读写操作。文本文件存储的优势在于简单、方便,适合存储少量信息,但想要规范的储存数据的话并不推荐使用文本文件储存。