爬虫(八):文件处理

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)

爬虫(八):文件处理

相关推荐