位置：首页-资讯-后端开发

【Python】爬虫练习-爬取豆瓣网电影评论用户的观影习惯数据

2023-10-25 12:40

短信预约 -IT技能 免费直播动态提醒

前言

一、配置环境

1.1、安装Python

1.2、安装Requests库和BeautifulSoup库

前言

豆瓣网是一个具有影响力的电影评论网站，其中包含大量的用户评论和评分数据。这些数据可以用于研究电影市场和用户观影习惯，同时还可以用于电影推荐算法以及在线视频网站的用户行为分析等方面，因此对于想要学习数据分析和机器学习的人来说，爬取豆瓣网电影评论数据是一个很好的练手项目。

本文将详细介绍如何使用Python爬虫抓取豆瓣网电影评论用户的观影习惯数据，并进行简单的数据处理和分析。

一、配置环境

在开始爬取豆瓣网电影评论数据之前，我们需要配置好Python开发环境和相关依赖库。

1.1、安装Python

首先，我们需要安装Python。建议使用Python 3.x版本。可以在官网下载并安装对应版本的Python：https://www.python.org/downloads/。

1.2、安装Requests库和BeautifulSoup库

接下来，我们需要安装两个Python库：Requests和BeautifulSoup。Requests库可以用于发送HTTP请求，获取网页内容，而BeautifulSoup库可以用于解析HTML文档，提取所需数据。

可以使用pip命令自动安装这两个库：

pip install requestspip install beautifulsoup4

1.3.、安装Matplotlib

如果我们需要对抓取的数据进行可视化分析，可以使用Matplotlib库。可以使用pip命令安装：

pip install matplotlib

安装完成后，我们就可以开始爬取豆瓣网电影评论数据了。

二、登录豆瓣网（重点）

豆瓣网对爬虫有一定的反爬虫策略，根据实战经验反爬一般作用于用户ID、用户IP和请求头的过滤，为了避免被封禁IP，代码中需要加入一些反爬虫措施：

设置请求间隔
使用代理IP
伪装请求头等

2.1、获取代理

使用代理可以帮助我们隐藏自己的真实IP地址，避免被封禁IP或者限制访问。在爬取豆瓣网电影评论数据时，如果频繁地访问同一个IP地址，豆瓣网可能会禁止我们的访问，因此使用代理也可以防止被封禁。这里我使用的是站大爷代理ip来进行操作的。

以下是使用用户名+密码授权获取ip代码示例：

 import requests#你要访问的网址url = 'http://www.ip138.com/'proxies = {        'http': 'http://账号:密码@ip:port',        'https': 'http://账号:密码@ip:port'#“用户名+密码”授权，账号为实例ID，密码为8位数字的密码，不用MD5加密,在后台的实例管理下可以看到    }response = requests.get(url, proxies=proxies)

2.2、测试代理ip是否可用

测试代理IP是否可用是为了确保我们使用的代理IP能够正常工作，防止由于代理IP失效或无效而导致爬虫请求失败。一般来说，我们可以通过发送请求来测试代理IP是否可用，如果请求成功则说明代理IP可用，否则说明代理IP无效或被封禁。

测试代理IP的代码可以通过如下方式实现：

定义一个函数，接收代理IP和目标网站URL作为参数
在函数中，使用requests库发送请求，并设置代理IP
如果请求成功，则说明代理IP可用，返回True
如果请求失败，则说明代理IP无效或被封禁，返回False

以下是一段测试代理IP的代码示例：

import requestsdef test_proxy_ip(proxy_ip, target_url):    proxies = {       'http': 'http://' + proxy_ip,        'https': 'http://' + proxy_ip    }    try:        response = requests.get(target_url, proxies=proxies, timeout=5)        if response.status_code == 200:            return True        else:            return False    except:        return False

在上述代码中，我们定义了一个test_proxy_ip函数，该函数接收代理IP和目标网站URL作为参数。在函数中，我们首先根据传入的代理IP构造proxies字典，然后使用requests库发送请求，并设置proxies参数为该字典。如果请求成功，则说明代理IP可用，函数返回True；否则函数返回False。

当我们在爬取豆瓣网电影评论数据时，可以使用上述代码测试代理IP是否可用。如果代理IP可用，则使用该代理IP发送请求；如果代理IP无效或被封禁，则需要更换新的代理IP。

2.3、设置大量请求头随机使用

爬取豆瓣网电影评论用户的观影习惯数据要设置大量请求头随机使用是为了防止网站监测到我们的请求头是模拟的，并防止被反爬虫机制封禁IP。因此，我们需要尽可能多的提供不同的请求头信息，对请求头进行随机使用，以降低被监测到的概率。

以下是一个随机使用请求头的示例代码：

import randomUSER_AGENTS = [  'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36 Edge/18.19582',  'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0',  'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36 Edge/16.16299',  # 更多用户代理信息...]def get_random_user_agent():    return random.choice(USER_AGENTS)headers = {   'User-Agent': get_random_user_agent(),   'Referer': 'https://www.douban.com',   'Cookie': 'your_cookie'}

在以上代码中，我们定义了一个USER_AGENTS列表，包含多个常用浏览器的User-Agent信息。然后定义了一个`get_random_user_agent()`函数，用于随机从USER_AGENTS列表中获取一个User-Agent信息。最后，我们设置headers变量，使用`get_random_user_agent()`函数获取随机的User-Agent信息，并将其赋值到headers变量中，作为请求头信息。

需要注意的是，我们在使用随机请求头的同时，仍然需要保持一定的爬取速度和请求间隔，以避免被封禁。同时，如果服务器在返回的内容中发现请求头信息异常，我们也可能被视为爬虫而被封禁。因此，在进行豆瓣网电影评论用户数据爬取时，需要谨慎处理请求头信息。

2.4、登录豆瓣网

以下是一个使用selenium模拟登录豆瓣网并保存cookie的示例代码：

from selenium import webdriverimport timeimport picklelogin_url = 'https://accounts.douban.com/passport/login?source=movie'username = 'your_username'password = 'your_password'cookie_file = 'cookies.pkl'# 启动浏览器driver = webdriver.Chrome()driver.maximize_window()driver.implicitly_wait(10)# 访问登录页面driver.get(login_url)# 输入用户名和密码driver.find_element_by_id('username').send_keys(username)driver.find_element_by_id('password').send_keys(password)# 等待用户输入验证码time.sleep(30)# 点击登录按钮driver.find_element_by_class_name('account-form-field-submit ').click()# 等待新页面加载完成time.sleep(10)# 保存cookie到文件cookies = driver.get_cookies()with open(cookie_file, 'wb') as f:    pickle.dump(cookies, f)# 关闭浏览器driver.quit()

在以上代码中，我们首先初始化了一些变量，包括登录页面的URL、豆瓣账号的用户名和密码、以及保存cookie的文件名。

然后，我们启动了Chrome浏览器，并通过`driver.get()`方法访问了豆瓣登录页面。接着，我们使用`selenium`模拟了用户在页面上输入了用户名和密码，并等待了30秒，以留出时间输入验证码。

最后，我们点击登录按钮，等待新页面加载完成，获取了cookies并将其保存到本地文件中。注意，这里我们使用了pickle模块将cookies对象序列化并保存到文件中，以便后续使用。

同时，我们还使用了`driver.quit()`方法关闭了浏览器。

三、爬取某一部热门电影数据

前面我们已经把爬取豆瓣网的准备工作全部完成，然后接下来就是爬取豆瓣网中某一部热门电影的各种数据了

3.1、爬取全部长、短评论

爬取热门电影的全部长短评论可以获得以下信息：

用户评价：通过阅读用户的评论，可以了解用户对电影的评价、看法和推荐指数等信息，有助于我们更好地了解电影的口碑和观众反应。
关键词提取：通过分析评论文本提取关键词，可以了解电影的主题、情感、人物关系等信息，有助于我们深入了解电影的故事情节和风格。
数据分析：通过统计评论数量、评分分布、用户画像等数据，可以帮助我们更好地了解电影的市场表现和受众群体，为影视公司、制片人和电影院等决策者提供有价值的数据支持。

因此，爬取热门电影的全部长短评论是一项非常有价值的工作，可以为我们提供大量有用的信息和数据分析基础。

以下是一个使用BeautifulSoup和requests库爬取豆瓣网某部电影全部长、短评论说明的示例代码

import requestsfrom bs4 import BeautifulSoup# 设置请求头headers = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.116 Safari/537.36'}# 豆瓣电影页面URLurl = 'https://movie.douban.com/subject/1292052/'# 发送请求获取页面HTMLresponse = requests.get(url, headers=headers)html = response.text# 使用BeautifulSoup解析HTMLsoup = BeautifulSoup(html, 'html.parser')# 查找长评和短评标签long_comments = soup.select_one('#hot-comments > div > div.article > span')short_comments = soup.select_one('#comments > div > h2 > span')# 提取评论数量并输出long_comments_count = long_comments.text.replace('全部', '').replace('条', '').strip()short_comments_count = short_comments.text.replace('全部', '').replace('条', '').strip()print(f"长评：{long_comments_count}")print(f"短评：{short_comments_count}")# 查找所有长评和短评内容long_comments_list = soup.select('#hot-comments > div > div.article > div > div.comment > p')short_comments_list = soup.select('#comments > div > div.comment > p')# 输出每个评论的内容for comment in long_comments_list:    print(comment.text.strip())   for comment in short_comments_list:    print(comment.text.strip())

在以上代码中，我们首先设置了请求头，指定了请求的User-Agent信息。然后，我们定义了电影页面的URL，并发送了GET请求获取了页面的HTML。

接着，我们使用BeautifulSoup解析了页面的HTML，并查找了长评和短评标签。通过提取标签文本中的评论数量信息，我们可以输出电影的长、短评数量。

最后，我们再次使用CSS选择器查找所有长、短评内容，并通过遍历列表输出了每个评论的文本内容。

3.2、获取用户观影数据

获取用户观影数据可以帮助我们了解用户的观影喜好和习惯，可以通过分析用户的观影历史和评分记录等内容，得到以下信息：

用户偏好：根据用户观影历史和评分记录，可以了解用户的观影偏好和喜好，有助于我们推荐更符合用户口味的电影和影视作品
影视推荐：通过分析用户的观影历史和评分记录，可以为用户推荐更符合其口味的电影和影视作品，提高用户体验和满意度
数据分析：通过统计用户评分和观影记录，可以了解影片的市场表现和受众群体，为制片人、电影院和影视公司等提供有价值的数据分析支持

因此，获取用户观影数据是一项非常有价值的工作。以下是获取用户观影数据的Python代码示例：

import requestsimport json# 登录豆瓣网session = requests.session()url = 'https://accounts.douban.com/j/mobile/login/basic'headers = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',    'Referer': 'https://accounts.douban.com/passport/login',}data = {    'name': 'your_username', # 替换为自己的用户名    'password': 'your_password', # 替换为自己的密码    'remember': 'false',}session.post(url, headers=headers, data=data)# 获取用户观影数据uid = 'your_user_id' # 替换为自己的用户IDurl = 'https://movie.douban.com/subject_suggest?q=%E7%94%B5%E5%BD%B1' # 电影页面URLheaders = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',    'Referer': 'https://www.douban.com/',}response = session.get(url, headers=headers)data = json.loads(response.text)movie_id = data[0]['id'] # 获取电影IDurl = f'https://movie.douban.com/subject/{movie_id}/' # 获取电影详情页面URLresponse = session.get(url, headers=headers)url = response.url + 'comments?start=0&limit=20&status=P&sort=new_score' # 获取电影短评页面URLresponse = session.get(url, headers=headers)data = response.textprint(data) # 输出评论内容

以上代码示例中，我们首先需要登录豆瓣网，然后通过获取电影ID和电影详情页面URL来获取其热门评论页面的URL，最后通过发送HTTP请求获取评论数据。注意，需要设置请求头信息，以避免请求被网站屏蔽。

3.3、存入数据库

将获取的数据存入数据库的好处：

数据持久化：将数据存入数据库中可以保证数据长期保存，避免数据丢失的情况
数据分析：将数据存入数据库后，可以使用SQL等工具进行数据分析和数据挖掘，得到更深入的信息和结论
数据管理：将数据存入数据库后，可以方便对数据进行管理、查询和更新

以下是将获取的去重后的数据存入数据库的Python代码示例：

import requestsimport jsonimport pymysql# 登录豆瓣网session = requests.session()url = 'https://accounts.douban.com/j/mobile/login/basic'headers = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',    'Referer': 'https://accounts.douban.com/passport/login',}data = {    'name': 'your_username', # 替换为自己的用户名    'password': 'your_password', # 替换为自己的密码    'remember': 'false',}session.post(url, headers=headers, data=data)# 获取用户观影数据uid = 'your_user_id' # 替换为自己的用户IDurl = 'https://movie.douban.com/subject_suggest?q=%E7%94%B5%E5%BD%B1' # 电影页面URLheaders = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',    'Referer': 'https://www.douban.com/',}response = session.get(url, headers=headers)data = json.loads(response.text)movie_id = data[0]['id'] # 获取电影IDurl = f'https://movie.douban.com/subject/{movie_id}/' # 获取电影详情页面URLresponse = session.get(url, headers=headers)url = response.url + 'comments?start=0&limit=20&status=P&sort=new_score' # 获取电影短评页面URLresponse = session.get(url, headers=headers)data = response.json()result = []for comment in data['comments']:    item = {        'id': comment['id'],        'user_name': comment['author']['name'],        'rating': comment['rating']['value'],        'create_time': comment['created_at'],        'content': comment['content'],    }    result.append(item)# 去重result2 = []for i in result:    if i not in result2:        result2.append(i)# 存入数据库db = pymysql.connect(host='localhost', user='root', password='123456', database='test')cursor = db.cursor()for item in result2:    sql = f"insert into movie_comment (id, user_name, rating, create_time, content) values ({item['id']}, '{item['user_name']}', {item['rating']}, '{item['create_time']}', '{item['content']}')"    cursor.execute(sql)db.commit()db.close()

以上代码示例中，我们使用Python连接到本地MySQL数据库，并将去重后的数据存入到数据库中的movie_comment表中。注意，需要根据自己的数据库用户名、密码、数据库名以及数据表名进行替换。

四、实战代码示例

由于豆瓣网对爬虫有一定的反爬虫策略，为了避免被封禁IP，代码中需要加入一些反爬虫措施，如设置请求间隔、使用代理IP、伪装请求头等。

以下是一个基于Python的爬虫示例，可以爬取指定电影的所有评论用户的观影习惯数据，并保存至CSV文件中：

import requestsfrom bs4 import BeautifulSoupimport csvimport timefrom random import randintMOVIE_URL = 'https://movie.douban.com/subject/xxxxxx/' # 请替换成具体电影的URLCSV_FILE = 'douban.csv' # CSV文件名# 设置请求头部信息，包括User-Agent和Refererheaders = {    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3',    'Referer': MOVIE_URL,}# 设置代理IP列表，用于反爬虫策略proxies = ['http://xxx.xxx.xx.xx:xxxx', 'http://yyy.yyy.yy.yy:yyyy', ...] # 请替换成实际的代理IP# 设置请求间隔，用于反爬虫策略WAIT_TIME = 1 # 单位为秒，可适当调整# 定义函数获取指定URL的HTML内容def get_html(url):    try:        proxy = {'http': proxies[randint(0, len(proxies)-1)]} # 从代理IP列表中随机选择一个代理        response = requests.get(url, headers=headers, proxies=proxy)        if response.status_code == 200:            return response.text    except requests.exceptions.RequestException as e:        print(e)        return None# 获取评论用户主页链接def get_user_link(comment_soup):    try:        user_tag = comment_soup.find('span', class_='comment-info').find('a')        user_link = user_tag.get('href')        return user_link    except:        return None# 解析评论用户主页，获取观影习惯数据def get_user_data(user_link):    user_data = {}    html = get_html(user_link)    time.sleep(WAIT_TIME)    if html:        soup = BeautifulSoup(html, 'html.parser')        # 获取用户常看类型        try:            type_tags = soup.find_all('span', class_='tags-body')            type_list = [t.text.strip() for t in type_tags]            user_data['type'] = ','.join(type_list)        except:            user_data['type'] = ''        # 获取用户评分前几名的电影（除当前电影外）        user_data['top_movies'] = ''        try:            movie_tags = soup.find_all('div', class_='info')            movie_list = []            for tag in movie_tags:                movie_name_tag = tag.find('a', class_='title')                if movie_name_tag and movie_name_tag['href'] != MOVIE_URL: # 排除当前电影                    movie_list.append(movie_name_tag.text)                if len(movie_list) >= 5: # 最多获取前5名                   break            user_data['top_movies'] = ','.join(movie_list)        except:            pass    return user_data# 主函数def main():    print('开始爬取...')    # 获取评论列表    comments = []    page = 0    while True:        page += 1        url = MOVIE_URL + 'comments?start=' + str((page-1) * 20)        html = get_html(url)        time.sleep(WAIT_TIME)        if html:            soup = BeautifulSoup(html, 'html.parser')            comment_tags = soup.find_all('div', class_='comment-item')            if comment_tags:               comments.extend(comment_tags)            else:                break # 没有更多评论        else:            break # 请求失败    # 获取每个评论用户的观影习惯数据    user_data_list = []    for comment in comments:        user_link = get_user_link(comment)        if user_link:            user_data = get_user_data(user_link)            user_data_list.append(user_data)            print('已爬取 {} 个用户观影数据'.format(len(user_data_list)))            time.sleep(WAIT_TIME)    # 将数据保存至CSV文件    with open(CSV_FILE, mode='w', encoding='utf-8', newline='') as f:        fieldnames = ['type', 'top_movies']        writer = csv.DictWriter(f, fieldnames=fieldnames)        writer.writeheader()        for data in user_data_list:            writer.writerow(data)    print('爬取完成，数据已保存至 {} 文件中。'.format(CSV_FILE))if __name__ == '__main__':    main()

需要注意的是，以上代码仅供参考，实际使用时需要根据具体情况进行调整。另外，爬虫涉及到的法律问题和道德问题需要认真考虑，不要用于非法用途或侵犯他人隐私等行为。

需要注意的几点：