前言

Python是一种跨平台的计算机程序设计语言。是一种面向对象的动态类型语言,最初被设计用于编写自动化脚本(shell),随着版本的不断更新和语言新功能的添加,越来越多被用于独立的、大型项目的开发。

日前,在学习中思考,如何整理一份贴吧历史数据出来,从中找寻当年玩贴吧的回忆呢。

于是就有了想从贴吧爬取数据的想法,此代码的由来就是如此。


事前准备

应先安装pycharm,此处使用的是社区版用于个人学习。如商业行为请支持正版pycharm套件。

检查需求

    获取贴吧列表的真实链接

    分析锁定回复量为筛选值,筛选出指定回复量以上的帖子

    将指定帖子的URL导出为表格形式保存

    通过URL读取帖子数据,保存一楼数据并存为word文件命名为帖子名(前5个字符).docx

开工

特别注意本文使用的是python3.7,无法直接从pycharm中add进docx包,否则会报错,请务必使用手动下载安装,原因及解决方法如下

原因:docx包中引用了exceptions模块,而python3.x版本移除了exceptions模块。即docx包尚未适配python3。

解决方法:

1.命令行 卸载docx,pip uninstall docx

2.下载 python_docx-0.8.10-py2.py3-none-any.whl     下载地址

3. 命令行输入pip install python_docx-0.8.10-py2.py3-none-any.whl使用新的docx包   

文件名应与下载的时候一致,也许之后会再进行更新

代码部分

获取链接模板及输出贴吧列表

打开浏览器,按下F12,打开贴吧,搜索一个贴吧,这边以"希灵帝国"为例

打开后可以看到,该帖吧首页的地址是:https://tieba.baidu.com/f?ie=utf-8&kw=%E5%B8%8C%E7%81%B5%E5%B8%9D%E5%9B%BD&fr=search

通过浏览器的翻译,我们可以得知:希灵帝国被浏览器翻译成%E5%B8%8C%E7%81%B5%E5%B8%9D%E5%9B%BD,所以这串其实就是希灵帝国四个字的unicode显示,所以得出了贴吧名字在贴吧链接里占用的模块。

该贴吧第四页的地址是:https://tieba.baidu.com/f?kw=%E5%B8%8C%E7%81%B5%E5%B8%9D%E5%9B%BD&ie=utf-8&pn=150

重新点回第一页:https://tieba.baidu.com/f?kw=%E5%B8%8C%E7%81%B5%E5%B8%9D%E5%9B%BD&ie=utf-8&pn=0

点第二页:https://tieba.baidu.com/f?kw=%E5%B8%8C%E7%81%B5%E5%B8%9D%E5%9B%BD&ie=utf-8&pn=50

得出规律:https://tieba.baidu.com/f?kw=贴吧名字&ie=utf-8&pn=页码

并且页码是0开始,50为一跳

所以这边可以定义读取的URL模板是:"https://tieba.baidu.com/f?kw=" + 贴吧名字 + "&ie=utf-8&pn={}"

{}是代表占位符,后续读取不同页面的时候就占用该占位符位置

然后我们就可以进行爬取页面数据的函数编写了

此处这个函数,主要功能是用来获取到贴吧列表页面的数据

# -*- coding:UTF-8 -*-
#设置间隔的
import time
#抓数据的
import requests
from bs4 import BeautifulSoup
#把数据弄能看的
import numpy as np
import pandas as pd
import docx
import xlrd
#上述是后面会用到的所有包,后续代码中不再另外列出


tiebaname = str(input("请输入贴吧名:"))   #定义贴吧名字,用户直接输入
template_url = "https://tieba.baidu.com/f?kw=" + tiebaname + "&ie=utf-8&pn={}"    #生成get请求的URL模板

def search_n_pages(n):
    #爬取n页的数据,确认爬取页数
    target = []

    #发起n次的get请求
    for i in range(n):
        #跟踪进度
        print('页数:',i+1)

        #按照浏览贴吧的自然行为,每一页50条
        target_url = template_url.format(50*(i))
        res = requests.get(target_url)

        #转为bs对象
        soup = BeautifulSoup(res.text,'html.parser')

        #获取该页帖子列表
        page_lst = soup.find_all(class_='j_thread_list clearfix')

        #该页信息保存到target里
        target.extend(extra_from_one_page(page_lst))

        #休息2秒再访问
        time.sleep(2)
    return target

分析筛选关键词及生成表单

然后有了贴吧页面的数据以后,就需要对页面数据进行筛选,筛选出需要的数据。

对应需求,筛选的关键词应是:回复量

所以这边,打开一个贴吧列表页面,对着回复量进行右键--审查元素可以找到对应的<div>块

image.png

找到class值   这边看到class值是  class="threadlist_rep_num center_text"    记录下来备用,待会抓取数据的时候就要使用到这个

再然后,要抓取对应的标题和URL,右键帖子标题,审查元素,就显示出了具体的细节

image.png

其中可以看到class值为class="j_th_tit ",而悄咪咪打开帖子对比一下,你会发现,帖子地址其实就是贴吧前缀+href的值  如图打开就是https://tieba.baidu.com/p/6297714220,其实就是https://tieba.baidu.com + a['href']

于是,回复量、标题、URL都获取到了,接下来就是将他们的数据爬取下来保存了

则这一段大概可以这么写:

# 设置回复量阈值  判断输入的回复量是否为int类型,不是的话重新要求输入
while True:
    try:
        clicknum = int(input("请输入要爬取的回复数(超过该回复数的才会被爬取):"))
        break
    except ValueError:
        print("请输入数字!")
M = clicknum
def extra_from_one_page(page_lst):
    '''从一页中提取 帖子'''
    # 临时列表保存字典数据,每一个帖子都是一个字典数据
    tmp = []
    for i in page_lst:
        #判断是否超过阈值(这边的“threadlist_rep_num”实际上是页面上显示的回复量)
        if int(i.find(class_='threadlist_rep_num center_text').text) > M:
            dic = {}
            #点击量
            dic['num'] = int(i.find(class_='col2_left j_threadlist_li_left').text)
            #帖子名称
            dic['name'] = i.find(class_='j_th_tit').text
            #帖子地址
            dic['address'] = 'https://tieba.baidu.com' + i.find(class_='j_th_tit').a['href']
            tmp.append(dic)
    return tmp
  
#爬取贴吧前n页数据 如果输入的不是int类型 则要求重新输入
while True:
    try:
        num = int(input("请输入要爬取的页数:"))
        break
    except ValueError:
        print("请输入数字!")
d = search_n_pages(num)

# 转化为pandas.DataFrame对象
data = pd.DataFrame(d)

# 导出到excel表格
xlsxname = tiebaname + '.xlsx'
data.to_excel(xlsxname)

这样就是生成了表格文件了

从表单中读取URL并生成保存为WORD文档

接下来是要从表格中读取URL,然后去打印到word里并保存文件

首先先读取上一步骤生成的表单'xlsxname',从中筛选出关键列'address',然后读取整列,并保存为列表(execl_list)

#从生成的表格中读取url列表,使url_write_word函数可以有读取用的url_list
def execl_read_url(execl_list):
    #打开指定表格
    rbook = xlrd.open_workbook(execl_list)
    rbook.sheets()
    rsheet = rbook.sheet_by_index(0)
    tmp = []
    #遍历表格每一行
    for row in rsheet.get_rows():
        #保留指定的列数数据
        product_column = row[3]
        product_value = product_column.value
        #去掉标题栏,将剩下的url保存为一个列表
        if product_value != 'address':
            r = str(row[3])
            r1 = r[6:-1]
            tmp.append(r1)
    return tmp
#读取表格,生成url列表
execl_list = xlsxname
#将url列表保存到url_list
url_list = execl_read_url(execl_list)

手中有一批的URL可以直接读取了,那接着要怎么去获取他们的页面里的数据呢?

当然是通过访问他们来获得数据。

打开随便一个帖子,对着标题进行审查元素,如下图

image.png

可以看到<div>块的class值是class="core_title_txt pull-left text-overflow  "   留存备用,后续可以用作文档标题和记录

而抓取数据的话,就先抓取一楼就好,对着一楼进行审查元素,如下图

image.png

抓取到的class值是class="d_post_content j_d_post_content "   留存备用

接着有了这两个值,就可以将网页数据中想要的部分筛选出来并保存了

调取之前生成的url列表,然后遍历列表的每个URL,逐一访问筛选出要保存的数据,通过docx库进行创建word文档并写入保存,这部分就完成了

#遍历url列表,访问并读取数据保存为word
def url_write_word(url_list):
    tmp = '文件生成完成'
    s = 1
    #遍历URL列表
    for i in url_list:
        tmp1 = []
        print('生成文件',s)
        s += 1
        i = str(i)
        #读取网页
        res = requests.get(i)
        soup = BeautifulSoup(res.text,'html.parser')
        url_title = soup.find(class_='core_title_txt pull-left text-overflow').text

        t1 = url_title
        t1 = t1[0:7] + '.docx'
        #创建一个word文档
        doc = docx.Document()
        url_word = soup.find(class_='d_post_content j_d_post_content').text
        t2 = url_word
        #将数据写入word
        doc.add_paragraph(i)
        doc.add_paragraph(t2)
        #保存文档
        doc.save(t1)
    return tmp

#生成word文档
url_write_word(url_list)


小结

本文使用了如下几个知识点:

  • 使用time模块进行程序间隔循环

  • 使用requests包进行get网页数据

  • 使用BeautifulSoup库进行整理筛选网页数据

  • 使用pandas包进行整理数据

  • 使用xlrd模块进行读取表格

  • 使用docx包进行创建保存文档

GitHub地址

参考资料

        用python爬取贴吧数据

        Python 读取excel指定的列

        python创建一个word文档并写入内容