Python爬取豆瓣+数据可视化

Python爬取豆瓣+数据可视化

博客原文和源码下载:Python爬取豆瓣+数据可视化

前言

前段时间应我姐邀请,看了一下Python爬虫。不得不说Python的语法确实简洁优美,可读性强,比较接近自然语言,非常适合编程的初学者上手。

在开始之前还是先介绍下什么是爬虫:

网络爬虫(英语:web crawler),也叫网络蜘蛛(spider),是一种用来自动浏览万维网的网络机器人。 — 维基百科

爬虫是一种程序或者脚本,用来替代人工浏览网页并从中提取信息,通常将提取的信息存储然后对其进行分析,从而获取有价值的信息。

爬虫也不是什么新鲜事物,可以说只要是编程语言几乎都能做到,而Python因为其简洁的语法和丰富的第三方库可以快速高效的编写爬虫,对于初学者可以说相当友好。接下来以爬取豆瓣电影TOP250页面为例说明如何用Python实现爬虫和数据可视化。

实现步骤

一、HTTP请求

导入第三方库requests,调用requests.get()方法向豆瓣电影TOP250页面发起GET请求,拿到响应的HTML。

二、数据提取

浏览器访问豆瓣电影TOP250页面并进入开发者模式复制要抓取节点的XPath。导入第三方库lxml的etree对象,调用etree.HTML(HTML)将HTML转换成element对象,然后使用element.xpath(XPath)方法拿到抓取的节点的文本(PS:也可以使用正则匹配文本)。

三、数据存储

导入第三方库openpyxl,调用openpyxl.Workbook()方法获取一个新的工作簿workbook,写入workbook.active=sheet获得一个工作表的引用sheet,为excel表指定位置添加数据:sheet[“A1”]=value,最后使用workbook.save(”excel表名“)保存数据,这里不用数据库是为了方便没有编程基础也能查看数据。

四、继续爬取

根据爬取页面的URL参数重复以上步骤,如豆瓣电影TOP250有两个参数start和filter,start表示页面的电影排行从TOP几的后一个开始,每个页面有25个电影,也就是说第一个页面参数start=0,第二个页面参数start=25,filter为过滤,暂时不用管,请求下一个页面时GET请求的参数start+25即可。

五、数据可视化

调用openpyxl.load_workbook(“excel表名”)方法获取存放数据的excel表,写入workbook.active=sheet获得一个工作表的引用,获取excel表指定位置的数据:data=sheet[“A1”].value。然后导入第三方库pyecharts,并根据文档调用合适的API生成图表实现数据可视化。

爬虫开发

创建一个项目Crawler,安装使用的第三方库:

pip install requests
pip install lxml
pip install openpyxl
pip install pyecharts

接下来在项目目录下新建文件html_parser.py:

from lxml import etree


class HTMLParser:  # HTML解析类
    def __init__(self):
        pass

    # 返回豆瓣电影Top250页面特定序号的电影的相关数据
    def parse(self, html, number):
        movie = {"title": "", "actors": "", "classification": "", "score": "", "quote": ""}
        selector = etree.HTML(html)
        movie["title"] = selector.xpath(
            "//*[@id="content"]/div/div[1]/ol/li[" + str(number) + "]/div/div[2]/div[1]/a/span[1]/text()")[0]
        movie["actors"] = selector.xpath(
            "//*[@id="content"]/div/div[1]/ol/li[" + str(number) + "]/div/div[2]/div[2]/p[1]/text()[1]")[0]
        movie["classification"] = selector.xpath(
            "//*[@id="content"]/div/div[1]/ol/li[" + str(number) + "]/div/div[2]/div[2]/p[1]/text()[2]")[0]
        movie["score"] = selector.xpath(
            "//*[@id="content"]/div/div[1]/ol/li[" + str(number) + "]/div/div[2]/div[2]/div/span[2]/text()")[0]
        # 如果存在则为该值,否则为空
        movie["quote"] = selector.xpath(
            "//*[@id="content"]/div/div[1]/ol/li[" + str(number) + "]/div/div[2]/div[2]/p[2]/span/text()")[0] if len(
            selector.xpath("//*[@id="content"]/div/div[1]/ol/li[" + str(
                number) + "]/div/div[2]/div[2]/p[2]/span/text()")) > 0 else ""
        return movie

该模块封装了解析豆瓣电影TOP250页面并提取数据的方法。

然后同样新建文件excel_handler.py:

import openpyxl


class ExcelHandler:  # excel文件处理类
    __book = None
    __sheet = None

    def __init__(self):
        self.book = None
        self.sheet = None
        pass

    # 获取工作簿并获得工作表的引用
    def startHandleExcel(self):
        self.book = openpyxl.Workbook()
        self.sheet = self.book.active

    # 为A、B、C、D、E列的指定行添加数据
    def handleExcel(self, row, A, B, C, D, E):
        self.sheet["A" + str(row)] = str(A).strip()
        self.sheet["B" + str(row)] = str(B).strip()
        self.sheet["C" + str(row)] = str(C).strip()
        self.sheet["D" + str(row)] = str(D).strip()
        self.sheet["E" + str(row)] = str(E).strip()
        return True

    # 处理完成保存excel
    def endHandleExcel(self, fileName):
        self.book.save(fileName)

    # 读取excel并获得工作表的引用
    def startReadExcel(self, fileName):
        self.book = openpyxl.load_workbook(fileName)
        self.sheet = self.book.active

    # 读取excel指定位置的数据
    def readExcel(self, coordinate):
        return str(self.sheet[coordinate].value)

    # 读取完成
    def endReadExcel(self):
        pass

该模块封装了存储和读取excel表的方法。

接下来在mian.py中写入:

import requests
import random
import time

from html_parser import HTMLParser
from excel_handler import ExcelHandler
from pyecharts import options as opts
from pyecharts.charts import Bar


def crawling():  # 爬取豆瓣电影Top250函数
    # 定义当前处理的excel表的行
    excelRow = 1
    # 实例化excel处理类
    excelHandler = ExcelHandler()
    # 开始处理excel
    excelHandler.startHandleExcel()
    # 在excel表中添加标题行
    excelHandler.handleExcel(excelRow, "名字", "演员", "分类", "评分", "引言")
    # 处理的行+1
    excelRow += 1

    # 定义豆瓣电影Top250页面的地址和所使用的user-agent(伪装为正常浏览器)
    url = "https://movie.douban.com/top250"
    headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) "
                             "Chrome/89.0.4389.82 Safari/537.36"}
    # 实例化网页解析类
    htmlParser = HTMLParser()
    print("开始爬取豆瓣电影Top250...")
    # 爬取豆瓣电影Top250的十个网页
    for page in range(10):
        # 定义URL的参数
        param = {"start": page * 25, "filter": ""}
        # 发起GET请求
        response = requests.get(url=url, params=param, headers=headers).text
        # 将请求结果页的25个电影信息存入excel
        for list in range(25):
            print("正在处理" + str(page + 1) + "页的第" + str(list + 1) + "部电影...")
            # 解析电影的信息
            movie = htmlParser.parse(response, list + 1)
            # 将解析结果存入excel
            excelHandler.handleExcel(excelRow, movie["title"], movie["actors"], movie["classification"], movie["score"],
                                     movie["quote"])
            # 处理的行+1
            excelRow += 1
        print("第" + str(page + 1) + "页爬取完成!")
        # 等待5-20秒之后再次爬取,模拟人的操作
        time.sleep(random.randint(5, 20))
    # excel存入完成
    excelHandler.endHandleExcel("movies.xlsx")
    print("豆瓣电影Top250爬取完成!")


def getCharts():  # 绘制评分数据图函数
    # 定义评分的字典
    scoreLevel = {}
    # 实例化excel处理类
    excelHandler = ExcelHandler()
    # 开始读取excel表
    excelHandler.startReadExcel("movies.xlsx")
    print("开始读取excel表中的评分列...")
    # 循环遍历excel表评分列
    for row in range(250):
        # 从excel表中读取评分列作为字典的key
        key = excelHandler.readExcel("D" + str(row + 2))
        # 如果该key存在则+1
        if key in scoreLevel:
            scoreLevel[key] += 1
        # 否则初始化该key的值为1
        else:
            scoreLevel[key] = 1
    # 读取excel结束
    excelHandler.endReadExcel()

    # 定义一个列表表示scoreLevel字典所有的key(即评分)
    keys = []
    # 定义一个列表表示scoreLevel字典所有的value(即该评分的数量)
    values = []
    # 提取scoreLevel的key和value
    for key in scoreLevel:
        keys.append(key)
        values.append(scoreLevel[key])
    print("评分数据读取完成!")

    print("开始绘制评分数据图...")
    # 绘制柱状图
    c = (
        Bar()
            .add_xaxis(keys)
            .add_yaxis("评分:数量(部)", values)
            .set_global_opts(
            title_opts=opts.TitleOpts(title="豆瓣电影评分Top250"),
            toolbox_opts=opts.ToolboxOpts(),
            legend_opts=opts.LegendOpts(is_show=False),
        )
            .render("movies_score.html")
    )
    print("评分数据图绘制完成!")


# 爬取豆瓣电影TOP250
crawling()

# 绘制评分数据图
getCharts()

main.py中的两个函数分别实现爬取豆瓣电影TOP250并存储数据和读取数据进行数据可视化操作,具体的过程已在注释中详细标注。

源码

源码下载访问博客原文:Python爬取豆瓣+数据可视化

相关链接

requests

lxml

openpyxl

pyecharts

Python 网络爬虫之Requests库的基本用法

python–lxml.etree解析html

Openpyxl 教程

最后

Python有很多优秀的爬虫框架,有兴趣可以自行了解,例如:

8个最高效的Python爬虫框架,你用过几个?

最后,爬虫不是什么内容都可以爬取,所谓”爬虫写的好,监狱进的早”,在爬取内容之前得先看一下:

  • 网络爬虫的法律规制

  • 网络爬虫?你可能面临的法律责任

  • 网络爬虫robots.txt的作用以及规范写法

等相关规范和限制。

hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » Python爬取豆瓣+数据可视化