import requests
from lxml import etree
import time
import csv

#信息頭
headers = {
 'User-Agent':'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.103 Safari/537.36'
}

#定義爬取函數(shù)
def douban_booksrank(url):
 res = requests.get(url, headers=headers)
 selector = etree.HTML(res.text)
 contents = selector.xpath('//div[@class="article"]/div[contains(@class,"doulist-item")]') #循環(huán)點(diǎn)
 for content in contents:
 try:
 title = content.xpath('div/div[2]/div[3]/a/text()')[0] #書名
 scores = content.xpath('div/div[2]/div[4]/span[2]/text()') #評(píng)分
 scores.append('9.0') #因?yàn)橛幸恍鴽](méi)有評(píng)分，導(dǎo)致列表為空，此處添加一個(gè)默認(rèn)評(píng)分，若無(wú)評(píng)分則默認(rèn)為9.0
 score = scores[0]
 comments = content.xpath('div/div[2]/div[4]/span[3]/text()')[0] #評(píng)論數(shù)量
 author = content.xpath('div/div[2]/div[5]/text()[1]')[0] #作者
 publishment = content.xpath('div/div[2]/div[5]/text()[2]')[0] #出版社
 pub_year = content.xpath('div/div[2]/div[5]/text()[3]')[0] #出版時(shí)間
 img_url = content.xpath('div/div[2]/div[2]/a/img/@src')[0] #書本圖片的網(wǎng)址
 img = requests.get(img_url) #解析圖片網(wǎng)址，為下面下載圖片
 img_name_file = 'C:/Users/lenovo/Desktop/douban_books/{}.png'.format((title.strip())[:3]) #圖片存儲(chǔ)位置，圖片名只取前3
 #寫入csv
 with open('C:\\Users\lenovo\Desktop\\douban_books.csv', 'a+', newline='', encoding='utf-8')as fp: #newline 使不隔行
 writer = csv.writer(fp)
 writer.writerow((title, score, comments, author, publishment, pub_year, img_url))
 #下載圖片，為防止圖片名導(dǎo)致格式錯(cuò)誤，加入try...except
 try:
 with open(img_name_file, 'wb')as imgf:
  imgf.write(img.content)
 except FileNotFoundError or OSError:
 pass
 time.sleep(0.5) #睡眠0.5s
 except IndexError:
 pass
#執(zhí)行程序
if __name__=='__main__':
 #爬取所有書本，共22頁(yè)的內(nèi)容
 urls = ['https://www.douban.com/doulist/1264675/?start={}&sort=time&playable=0&sub_type='.format(str(i))for i in range(0,550,25)]
 #寫csv首行
 with open('C:\\Users\lenovo\Desktop\\douban_books.csv', 'a+', newline='', encoding='utf-8')as f:
 writer = csv.writer(f)
 writer.writerow(('title', 'score', 'comment', 'author', 'publishment', 'pub_year', 'img_url'))
 #遍歷所有網(wǎng)頁(yè)，執(zhí)行爬取程序
 for url in urls:
 douban_booksrank(url)

from matplotlib import pyplot as plt
import pandas as pd
import re

plt.rcParams['font.sans-serif']=['SimHei'] #用來(lái)正常顯示中文標(biāo)簽
plt.rcParams['axes.unicode_minus']=False #用來(lái)正常顯示負(fù)號(hào)
plt.subplots_adjust(wsapce=0.5, hspace=0.5) #調(diào)整subplot子圖間的距離

pd.set_option('display.max_rows', None) #設(shè)置使dataframe 所有行都顯示

df = pd.read_csv('C:\\Users\lenovo\Desktop\\douban_books.csv') #讀取csv文件，并賦為dataframe對(duì)象

comment = re.findall('\((.*?)人評(píng)價(jià)', str(df.comment), re.S) #使用正則表達(dá)式獲取評(píng)論人數(shù)
#將comment的元素化為整型
new_comment = []
for i in comment:
 new_comment.append(int(i))

pub_year = re.findall(r'\d{4}', str(df.pub_year),re.S) #獲取書籍出版年份
#同上
new_pubyear = []
for n in pub_year:
 new_pubyear.append(int(n))

#繪圖
#1、繪制書籍評(píng)分范圍的直方圖
plt.subplot(2,2,1)
plt.hist(df.score, bins=16, edgecolor='black')
plt.title('豆瓣書籍排行榜評(píng)分分布', fontweight=700)
plt.xlabel('scores')
plt.ylabel('numbers')

#繪制書籍評(píng)論數(shù)量的直方分布圖
plt.subplot(222)
plt.hist(new_comment, bins=16, color='green', edgecolor='yellow')
plt.title('豆瓣書籍排行榜評(píng)價(jià)分布', fontweight=700)
plt.xlabel('評(píng)價(jià)數(shù)')
plt.ylabel('書籍?dāng)?shù)量（單位/本）')

#繪制書籍出版年份分布圖
plt.subplot(2,2,3)
plt.hist(new_pubyear, bins=30, color='indigo',edgecolor='blue')
plt.title('書籍出版年份分布', fontweight=700)
plt.xlabel('出版年份/year')
plt.ylabel('書籍?dāng)?shù)量/本')

#尋找關(guān)系
plt.subplot(224)
plt.bar(new_pubyear,new_comment, color='red', edgecolor='white')
plt.title('書籍出版年份與評(píng)論數(shù)量的關(guān)系', fontweight=700)
plt.xlabel('出版年份/year')
plt.ylabel('評(píng)論數(shù)')

plt.savefig('C:\\Users\lenovo\Desktop\\douban_books_analysis.png') #保存圖片
plt.show()