1. 程式人生 > >數據挖掘_多線程抓取

數據挖掘_多線程抓取

任務 打印 會有 time() gda ads .... 方案 multipl

在這一篇文章中,我們主要來介紹多線程抓取數據。

技術分享圖片

多線程是以並發的方式執行的,在這裏要註意,Python的多線程程序只能運行在一個單核上以並發的方式運行,即便是多核的機器,所以說,使用多線程抓取可以極大地提高抓取效率

下面我們以requests為例介紹多線程抓取,然後在通過與單線程程序比較,體會多線程的效率的提高

這一次,我就不用我的網站做測試了,因為網站的內容此時還並不是太多,不能體現多線程的優勢

我們通過當當網來測試我們的多線程實例,通過對搜索結果的同一抓取實現功能的演示,搜索模式地址如下

http://search.dangdang.com/?key=Python&act=input&page_index=1

 可以看到key代表的是搜索關鍵字,act代表你是通過什麽方式搜索的,page_index代表的是搜索頁面的頁碼

技術分享圖片

在抓取到上面的頁碼後,要對裏面的信息進行提取,最後將提取的信息保存在文本文件中,文件中保存每一本書的書名,以及他的鏈接

下面我們定義抓取實驗所需要的方法(或函數)

# coding=utf-8
__Author__ = "susmote"

import requests
from bs4 import BeautifulSoup


def format_str(s):
    return s.replace("\n", "").replace(" ", "").replace("\t", "")

def get_urls_in_pages(from_page_num, to_page_num):
    urls = []
    search_word = "python"
    url_part_1 = "http://search.dangdang.com/?key="
    url_part_2 = "&act=input"
    url_part_3 = "&page_index="
    for i in range(from_page_num, to_page_num + 1):
        urls.append(url_part_1 + search_word + url_part_2 + url_part_3 + str(i))
    all_href_list = []
    for url in urls:
        print(url)
        resp = requests.get(url)
        bs = BeautifulSoup(resp.text, "lxml")
        a_list = bs.find_all("a")
        needed_list = []
        for a in a_list:
            if ‘name‘ in a.attrs:
                name_val = a[‘name‘]
                href_val = a[‘href‘]
                title = a.text
                if ‘itemlist-title‘ in name_val and title != "":
                    if [title, href_val] not in needed_list:
                        needed_list.append([format_str(title), format_str(href_val)])
        all_href_list += needed_list
    all_href_file = open(str(from_page_num) + ‘_‘ + str(to_page_num) + ‘_‘ + ‘all_hrefs.txt‘, ‘w‘)
    for href in all_href_list:
        all_href_file.write(‘\t‘.join(href) + ‘\n‘)
    all_href_file.close()
    print(from_page_num, to_page_num, len(all_href_list))

  下面我們來解釋一下這些代碼

首先,format_str是用來在提取信息後去掉多余的空白

而get_url_in_pages方法是執行功能的主體,這個方法接收的參數是指頁碼的範圍,函數體中,urls主要用來存放基於兩個參數所生成的所有要抓取的頁面的鏈接,我把url分為3個部分,也是為了方便之後對鏈接的組合,然後的for循環就是做的是拼接的工作,這裏我不多做解釋了,如果不懂,請留言

下一步,我們定義了一個列表all_href_list ,這個列表是用來存儲每頁中包含圖書信息的,實際上它又是一個嵌套的列表,裏面的元素是[書名, 鏈接],他的形式如下所示

all_href_list = [
    [‘書名1‘, "鏈接1"],
    [‘書名2‘, "鏈接2"],
    [‘書名3‘, "鏈接3"],
    ......
]

後面的代碼就是對滅一頁進行抓取和提取信息了,這部分的代碼都在for url in urls這個循環體中,首先打印鏈接,然後調用requs的get方法,獲取頁面,之後又使用BeautifulSoup將get請求放回的HTML文本進行分析,轉為BeautifulSoup能夠處理的結構,命名為bs

之後我們定義的needed_list是用來存放書名和鏈接的,bs.find_all(‘a‘)抽取了頁面中所有的鏈接元素,for a in a_list 對每一個列表中的元素進行遍歷分析,在這之前,我們通過瀏覽器發現了他的結構

技術分享圖片

每個書籍元素中都會有一個屬性name,值為"itemlist_title",通過這個我們很容易的就篩選出了書籍元素,然後將書籍信息,以及鏈接元素href一同存入列表,在存之前,我們還做了一些判斷,是否已經存在這個鏈接了,和這個元素的鏈接為空

每抽取完一個頁面的鏈接後,就可以把它加入到all_href_list中,也就是下面這行代碼

all_href_list += needed_list

  註意我在這裏使用的是 += 運算符

獲取到範圍內所有的鏈接元素後,就可以寫入文件了,在這裏我不做過多解釋了

然後我們下一步就是定義多線程了,因為我們搜索關鍵詞總的頁數是32頁

技術分享圖片

所以我們在這裏準備用3個線程來完成這些任務,也就是每個線程處理10個頁面,在單線程的情況下,這30頁用一個線程單獨完成

下面我們給出抓取方案的代碼

# coding=utf-8
__Author__ = "susmote"

import time
import threading
from mining_func import get_urls_in_pages

def multiple_threads_test():
    start_time = time.time()
    page_range_list = [
        (1, 10),
        (11, 20),
        (21, 32),
    ]
    
    th_list = []
    for page_range in page_range_list:
        th = threading.Thread(target = get_urls_in_pages, args = (page_range[0], page_range[1]))
        th_list.append(th)
    
    for th in th_list:
        th.start()
        
    for th in th_list:
        th.join()
        
    end_time = time.time()
    print("共使用時間1:", end_time - start_time)
    return end_time - start_time

  

簡單解釋一下,為了獲取運行的時間,我們定義了一個開始時間start_time 和 一個結束時間end_time,運行時間也就是結束時間減去開始時間

然後定義的一個列表page_range_list也就是把頁碼分為三段,前面有提到過

之後又定義了一個列表th_list也就是存儲所有線程對象的列表,之後通過一個循環,生成了3個線程對象,分別對應著不同的頁碼範圍,把他們存入列表

然後在後面的循環中,分別執行th.start(),開啟線程,在後面,我們為了使這些異步並發執行的線程都執行完畢後再退出函數,這裏使用了線程的join方法,等待各線程執行完畢

下面就是最激動人心的時候了,對代碼進行測試

在這裏,我們寫下如下代碼

# coding=utf-8
__Author__ = "susmote"

from mining_threading import multiple_threads_test

if __name__ == "__main__":
    mt = multiple_threads_test()
    print(‘mt‘, mt)

  

為了使測試結果更加精確,我們進行三次實驗,取平均時間

第一次實驗

技術分享圖片

使用時間6.651

第二次實驗

技術分享圖片

使用時間6.876

第三次實驗

技術分享圖片

使用時間6.960

平均時間如下

技術分享圖片

6.829

下面是單進程代碼

# coding=utf-8
__Author__ = "susmote"

import time
from mining_func import get_urls_in_pages


def sigle_test():
    start_time = time.time()
    get_urls_in_pages(1, 32)
    end_time = time.time()
    print("共使用時間 : ", end_time - start_time)
    return end_time - start_time 

  

調用函數如下

# coding=utf-8
__Author__ = "susmote"

from single_mining import single_test


if __name__ == "__main__":
    st = single_test()
    print(‘st ‘, st) 

  

在命令行下執行

第一次

技術分享圖片

10.138

第二次

技術分享圖片

10.290

第三次

技術分享圖片

10.087

平均花費時間

技術分享圖片

10.171

所以說,多線程的確能夠提高抓取的效率,註意,這是在數據比較少的情況進行的,如果數據量比較大的話,多線程的優勢就很明顯了

你可以自己去更改搜索關鍵詞,和頁碼,或是重新找一個網頁(抓取跟網速也有很大的關系)

附幾張抓取數據的圖

技術分享圖片

技術分享圖片

數據挖掘_多線程抓取