爬虫(八)：文件处理

1. json文件处理

1.1 什么是json

JSON(JavaScript Object Notation，JS对象简谱)是一种轻量级的数据交换格式。它基于ECMAScript(欧洲计算机协会制定的js规范)的一个子集，采用完全独立于编程语言的文本格式来存储和表示数据。简洁和清晰的层次结构使得JSON成为理想的数据交换语言。 易于人阅读和编写，同时也易于机器解析和生成，并有效地提升网络传输效率。

1.2 json支持数据格式

1.对象（字典）。使用花括号。

2.数组（列表）。使用方括号。

3.整型、浮点型、布尔类型和null类型。

4.字符串类型（字符串必须要用双引号，不能用单引号）。

多个数据之间使用逗号分开。

注意：json本质上就是一个字符串。

1.3 字典和列表转json

import json

books = [
    {
        "title":"坏蛋是怎样炼成的1","price":9.8
    },
    {
        "title":"坏蛋是怎样炼成的2","price":9.9
    }
 ]

json_str = json.dumps(books,ensure_ascii=False)
print(json_str)
print(type(books))
print(type(json_str))

因为json在dump的时候，只能触发ascii的字符，因此会将中文进行转义，这时候我们可以使用ensure_ascii=False关闭这个特性。

在python中，只有基本数据类型才能转换成json格式的字符串。即：int、float、str、list、dict、tuple。

将json数据直接dump到文件中：

json模块中处理dumps函数，还有一个dump函数，这个函数可以传入一个文件指针，直接将字符串dump到文件中。

import json

books = [
    {
        "title":"坏蛋是怎样炼成的1","price":9.8
    },
    {
        "title":"坏蛋是怎样炼成的2","price":9.9
    }
 ]

with open('a.json','w',encoding='utf-8') as fp:
    json.dump(books,fp,ensure_ascii=False)

dump和dumps的区别：

dump()不需要使用write()方法，只需要写哪个字典、哪个文件即可；而.dumps()需要使用write()方法写入

如果要把字典写到文件里面的时候，dump()好用；但如果不需要操作文件，或需要把内容存到数据库和Excel，则需要使用dumps()先把字典转成字符串，再写入。

1.4 将一个json字符串load成python对象

#encoding: utf-8
import json

json_str = '[{"title": "坏蛋是怎样炼成的1", "price": 9.8}, {"title": "坏蛋是怎样炼成的2", "price": 9.9}]'

books = json.loads(json_str)
print(type(books))
for book in books:
    print(book)

直接从文件中读取json：

#encoding: utf-8
import json

with open('a.json','r',encoding='utf-8') as fp:
   books = json.load(fp)
   for book in books:
       print(book)

load与loads的区别：

loads()传的是字符串，而load()传的是文件对象

使用loads()时需要先读文件再使用，而load()则不用

2. CSV文件处理

CSV是一种通用的、相对简单的文件格式，被用户、商业和科学广泛应用。最广泛的应用是在程序之间转移表格数据，而这些程序本身是在不兼容的格式上进行操作的（往往是私有的和/或无规范的格式）。因为大量程序都支持某种CSV变体，至少是作为一种可选择的输入/输出格式。

2.1 读取csv文件

csv数据：

csv文件不能直接创建，只能由excel文件另存为csv格式。

import csv

with open('Demo.csv','r') as fp:
    reader = csv.reader(fp)
    titles = next(reader)
    for x in reader:
        print(x)

 

为什么读取不到第一行呢？

因为第一行一般都是存放标题等信息。 

这样操作，以后获取数据的时候，就要通过上面的方法来获取数据。 如果想要在获取数据的时候通过标题来获取，那么可以用DictReader。

import csv

with open('Demo.csv','r') as fp:
    reader = csv.DictReader(fp)
    for x in reader:
        print(x)

 

我们可以通过标题来选中返回的数据：

import csv

with open('Demo.csv','r') as fp:
    reader = csv.DictReader(fp)
    for x in reader:
        print(x['数据1'])

 

2.2 写入数据到csv文件

写入数据到csv文件，需要插件一个writer对象，主要用到两个方法。一个是writerow，这个是写入一行。一个是writerows，这个是写入多行。

import csv

headers = {'name','age','height'}
values = {
    ('张三',18,170),
    ('李四',20,175),
    ('王五',21,180)
}
with open('test.csv','w',encoding='utf-8') as fp:
    writer = csv.writer(fp)
    writer.writerow(headers)
    writer.writerows(values)

也可以使用字典的方式把数据写入。这时候就需要使用DictWriter了。

import csv

headers = ['name','age','height']
values = [
    {'name':'张三','age':18,'height':170},
    {'name':'李四','age':20,'height':175},
    {'name':'王五','age':21,'height':180}
]
with open('test.csv','w',encoding='utf-8') as fp:
    writer = csv.DictWriter(fp,headers)
    writer.writeheader()
    writer.writerows(values)