完美假期第一步:用Python尋找更便宜的航班!
大數(shù)文摘出品
編譯:高延、熊琰、胡笳、蔣寶尚
“你喜歡旅游嗎?”
這個(gè)簡(jiǎn)單的問(wèn)題經(jīng)常會(huì)得到一個(gè)積極的回復(fù)甚至還會(huì)額外收到一個(gè)或兩個(gè)冒險(xiǎn)的故事。通常來(lái)講,旅行是一種體驗(yàn)新文化和拓寬自己視野的好方法。
但如果把問(wèn)題換成“你喜歡查機(jī)票的過(guò)程嗎?”,我敢肯定大家的反應(yīng)一定會(huì)不那么熱情......
那么,用Python解決你的難點(diǎn)吧!本文作者Fábio Neves,一位資深的商業(yè)數(shù)據(jù)分析師將會(huì)帶你建立一個(gè)網(wǎng)絡(luò)爬蟲(chóng)項(xiàng)目,幫助我們找到最優(yōu)惠的價(jià)格!
具體做法是對(duì)特定目的地以及靈活的日期范圍(根據(jù)你選擇的日期前后最多3天)進(jìn)行航班價(jià)格搜索。
搜索結(jié)果保存到一個(gè)excel中并為你發(fā)送一封展示快速統(tǒng)計(jì)信息的電子郵件。顯然,最終的目的是幫助我們找到最優(yōu)惠的價(jià)格!
如果你真的想試試,你可以在服務(wù)器上執(zhí)行這個(gè)腳本(一個(gè)簡(jiǎn)單的Raspberry Pi就可以(注:Raspberry Pi樹(shù)莓派又稱卡片式電腦,外形只有信用卡大小,運(yùn)算性能和智能手機(jī)相仿。So大家在自己筆記本電腦上折騰就足夠了。。)),并且每天運(yùn)行一次或兩次。它會(huì)將檢索結(jié)果以郵件的形式發(fā)給你,我建議將excel文件保存到Dropbox云端,這樣你就可以隨時(shí)隨地訪問(wèn)它。
注:Dropbox是一個(gè)類(lèi)似于百度云的云端服務(wù)
我還是沒(méi)找到任何錯(cuò)誤低價(jià)票,但我想還是有可能的!
它會(huì)根據(jù)“彈性日期范圍”進(jìn)行檢索,以便查找你首選日期的前后最多3天的所有航班。盡管該腳本一次只適用于一組from/to目的地,但你可以輕松地調(diào)整它在每個(gè)循環(huán)內(nèi)運(yùn)行多組行程目的地。你甚至可能最終找到一些錯(cuò)誤低價(jià)票......簡(jiǎn)直棒極了!
爬蟲(chóng)腳本
當(dāng)我第一次開(kāi)始做網(wǎng)絡(luò)爬蟲(chóng)時(shí),我對(duì)這塊并不特別感興趣。我本想用預(yù)測(cè)建模,財(cái)務(wù)分析和一些情緒分析來(lái)做更多的項(xiàng)目,但事實(shí)證明,弄明白如何構(gòu)建第一個(gè)網(wǎng)絡(luò)爬蟲(chóng)是很有趣的。隨著我不斷學(xué)習(xí),我意識(shí)到網(wǎng)絡(luò)抓取是互聯(lián)網(wǎng)運(yùn)轉(zhuǎn)的精髓。
是的......就像Larry 和 Sergey一樣,在啟動(dòng)爬蟲(chóng)程序后去盡情享受按摩浴缸吧!
你可能認(rèn)為這是一個(gè)非常大膽的想法,但如果我告訴你谷歌就誕生于Larry 和 Sergey通過(guò)Java和Python寫(xiě)的爬蟲(chóng)程序呢?谷歌通過(guò)爬取整個(gè)互聯(lián)網(wǎng)來(lái)試圖為你的問(wèn)題提供最佳答案。有非常多關(guān)于網(wǎng)絡(luò)爬蟲(chóng)的應(yīng)用程序,即便你更喜歡數(shù)據(jù)科學(xué)中的其他主題,你仍然需要一些爬蟲(chóng)技能來(lái)獲取想要的數(shù)據(jù)。
Python可以來(lái)拯救你
第一個(gè)挑戰(zhàn)是選擇從哪個(gè)平臺(tái)抓取信息。這其實(shí)并不容易,但我最終選擇了Kayak。決定之前我嘗試了Momondo,Skyscanner,Expedia等等,但這些網(wǎng)站上的驗(yàn)證碼部分真的是讓人抓狂。經(jīng)過(guò)幾次嘗試選擇交通信號(hào)燈,人行橫道和自行車(chē)的這種“你是真人嗎”的檢查后,我的結(jié)論是Kayak是目前最好的選擇,即使它在短時(shí)間內(nèi)加載太多頁(yè)面時(shí)也會(huì)拋出安全性校驗(yàn)。
我設(shè)置機(jī)器人以4到6小時(shí)的間隔來(lái)查詢網(wǎng)站,這樣就不會(huì)有問(wèn)題了。在這里和那里偶爾可能會(huì)出現(xiàn)卡殼中斷現(xiàn)象,但是如果你遇到驗(yàn)證碼校驗(yàn),那么你需要手動(dòng)進(jìn)行驗(yàn)證碼認(rèn)證,確認(rèn)完畢后再啟動(dòng)機(jī)器人程序,然后等待幾個(gè)小時(shí)它就會(huì)重置。你也可以隨意將這些代碼應(yīng)用到其他平臺(tái),歡迎你在評(píng)論部分分享你的應(yīng)用!
如果你是個(gè)爬蟲(chóng)新手,或者還不了解為什么有一些網(wǎng)站總會(huì)設(shè)置各種障礙來(lái)阻止網(wǎng)絡(luò)抓取,那么在寫(xiě)第一行爬蟲(chóng)代碼之前,請(qǐng)你先閱讀了解一下谷歌“ 網(wǎng)絡(luò)抓取禮儀 ”。如果你像瘋子一樣準(zhǔn)備好了開(kāi)始網(wǎng)絡(luò)抓取,你獲得努力成果可能會(huì)比你想象的要快得多。
網(wǎng)絡(luò)抓取禮儀 :http://lmgtfy.com/?q=web+scraping+etiquette
請(qǐng)系好安全帶……
打開(kāi)chrome標(biāo)簽頁(yè)后,我們將定義一些在循環(huán)內(nèi)使用的函數(shù)。關(guān)于整體結(jié)構(gòu)的大致想法是這樣的:
- 一個(gè)函數(shù)將啟動(dòng)機(jī)器人,聲明我們想要搜索的城市和日期。
- 此功能獲取第一批搜索結(jié)果并按“最佳”航班進(jìn)行排序,隨后點(diǎn)擊“加載更多結(jié)果”。
- 另一個(gè)函數(shù)將抓取整個(gè)頁(yè)面,并會(huì)返回一個(gè)dataframe數(shù)據(jù)集
- 重復(fù)步驟2和3獲取“最便宜”和“最快”的排序結(jié)果。
- 電子郵件將價(jià)格的最終結(jié)果(最便宜和平均值)發(fā)送給你,并且將三個(gè)排序(價(jià)格、時(shí)間、整體最佳)的數(shù)據(jù)集保存為一個(gè)excel文件
- 前面的所有步驟循環(huán)重復(fù),每隔X小時(shí)運(yùn)行一次。
OK,每個(gè)Selenium項(xiàng)目都將以webdriver作為開(kāi)頭。我用的是ChromeDriver,當(dāng)然還有其他選擇。比如,PhantomJS或Firefox也很受歡迎。webdriver下載好之后,將其放在一個(gè)文件夾中就可以了。代碼的第一行將會(huì)自動(dòng)打開(kāi)一個(gè)空白的Chrome標(biāo)簽頁(yè)。
請(qǐng)注意,我不是在這里開(kāi)辟新天地,或是提出一種非常具有開(kāi)拓性的創(chuàng)新。當(dāng)下確實(shí)已經(jīng)有更先進(jìn)的方法來(lái)尋找便宜的票價(jià),但我希望我的這個(gè)帖子可以跟大家分享一些簡(jiǎn)單而實(shí)用的東西!
這些是我用于整個(gè)項(xiàng)目所引用的包。我將使用randint來(lái)讓機(jī)器人在每次搜索之間隨機(jī)停頓幾秒鐘。這是所有機(jī)器人所必備的功能。如果你運(yùn)行了前面的代碼,則需要先打開(kāi)一個(gè)Chrome網(wǎng)頁(yè)窗口作為機(jī)器人檢索的入口。
所以,先讓我們來(lái)快速測(cè)試一下,在新網(wǎng)頁(yè)打開(kāi)kayak.com。選擇你要飛往的城市和日期。選擇日期時(shí),請(qǐng)務(wù)必選擇“+ -3天”。我已經(jīng)編寫(xiě)了相關(guān)的代碼,如果你只想搜索特定日期,那么你需要適當(dāng)?shù)剡M(jìn)行一些調(diào)整。我將盡量在整個(gè)文本中指出所有的變動(dòng)值。
點(diǎn)擊搜索按鈕并獲取地址欄中的鏈接。這個(gè)鏈接應(yīng)該就是我在下面需要用的鏈接,在這里我將變量kayak定義為url并調(diào)用webdriver的get方法。你的搜索結(jié)果接下來(lái)應(yīng)該就會(huì)出現(xiàn)了。
每當(dāng)短時(shí)間內(nèi)多次使用get命令的時(shí)候,系統(tǒng)就會(huì)跳出驗(yàn)證碼檢查。你可以手動(dòng)解決驗(yàn)證碼問(wèn)題,并在下一個(gè)問(wèn)題出現(xiàn)之前繼續(xù)測(cè)試腳本。從我的測(cè)試來(lái)看,第一次搜索運(yùn)行似乎一切正常,所以如果你想要用這段代碼,并且讓它們之間保持較長(zhǎng)的執(zhí)行間隔,就可以解決掉這個(gè)難題。你并不需要每10分鐘就更新這些價(jià)格,不是嗎?!
XPath的坑
目前為止,我們打開(kāi)了一個(gè)瀏覽器窗口并獲得了網(wǎng)址。接下來(lái)我會(huì)使用XPath或者CSS選擇器來(lái)抓取價(jià)格等其他信息。曾經(jīng)我也只用XPath,當(dāng)時(shí)我覺(jué)得沒(méi)必要用CSS,但是現(xiàn)在看來(lái)最好結(jié)合著用。你可以直接用瀏覽器復(fù)制網(wǎng)頁(yè)XPath來(lái)用,你也會(huì)發(fā)現(xiàn)由XPath雖可以定位網(wǎng)頁(yè)元素但是可讀性很差,所以我漸漸意識(shí)到只用XPath很難獲得你想要的頁(yè)面元素。有時(shí)候,指向得越細(xì)就越不好用。
接下來(lái),我們用Python來(lái)選擇出最低票價(jià)的頁(yè)面元素。上述代碼中紅色部分就是XPath選擇器的代碼,在網(wǎng)頁(yè)中,你可以在任意位置點(diǎn)擊右鍵并選擇“檢查”來(lái)找到它。試試吧,在你想看代碼的地方點(diǎn)右鍵,“檢查”它。
為了說(shuō)明我前面所說(shuō)的XPath的不足,請(qǐng)大家對(duì)比如下差異:
- 1 # This is what the copy method would return. Right click highlighted rows on the right side and select "copy > Copy XPath"
- //*[@id="wtKI-price_aTab"]/div[1]/div/div/div[1]/div/span/span
- 2 # This is what I used to define the "Cheapest" button
- cheap_results = ‘//a[@data-code = “price”]’
上述代碼中,第二種方式的簡(jiǎn)潔性清晰可見(jiàn)。它會(huì)去搜素具有data-code屬性值為price的a元素。而第一種方式則是去搜素一個(gè)id為wtKI-price_aTab元素,且該元素嵌在5層div及2層span內(nèi)。對(duì)于這次頁(yè)面,它能起作用,但這里的坑在于,下次加載頁(yè)面時(shí),這個(gè)id會(huì)變,而且每次加載時(shí)wtKI值也是動(dòng)態(tài)變化的,所以到時(shí)候這段代碼就無(wú)效了。所以多花點(diǎn)功夫研究一下XPath表示的內(nèi)容還是對(duì)你有價(jià)值的。
不過(guò)這種直接復(fù)制XPath的方法對(duì)于那些不是很復(fù)雜善變的頁(yè)面來(lái)說(shuō)還是蠻好用的。
基于上述代碼結(jié)果,如果我想找出所有匹配的結(jié)果并存到list里,該怎么做呢?很簡(jiǎn)單,因?yàn)樗薪Y(jié)果都在CSS對(duì)象resultWrapper中,只要按照我下圖代碼中寫(xiě)個(gè)for循環(huán)就能獲得所有結(jié)果。這個(gè)思路掌握了,那下圖的代碼你也就基本看明白了。也就是說(shuō),先選定最外層的頁(yè)面元素(如本文網(wǎng)站中的resultWrapper),再找一種方式(如XPath)來(lái)獲取信息,最后再將信息存到可讀的對(duì)象中(本例中先存在flight_containers中,再存在flights_list中)。
我把前三個(gè)結(jié)果詳細(xì)內(nèi)容都打印出來(lái)了,里面有我們需要的全部有用信息,但我們還是要找個(gè)更好的方法提取它們,這時(shí)我們就要對(duì)這些元素單獨(dú)解析。
開(kāi)始爬數(shù)據(jù)!
最簡(jiǎn)單的代碼就是讀取更多這個(gè)函數(shù),我們先從這里開(kāi)始。我希望在不觸發(fā)安全校驗(yàn)的情況下獲取盡量多的航班,所以在每次加載完頁(yè)面我都會(huì)點(diǎn)擊“load more results”按鈕。值得注意的是我用到了try語(yǔ)句,因?yàn)橛械臅r(shí)候不一定會(huì)存在這個(gè)按鈕。
哦嘞,前期鋪墊的有點(diǎn)長(zhǎng)(抱歉,我確實(shí)比較容易跑偏)。我們現(xiàn)在要開(kāi)始定義用于爬數(shù)據(jù)的函數(shù)了。
我在下文會(huì)提到的page_scrape函數(shù)中解析了大部分元素。有時(shí)候,返回來(lái)的航班l(xiāng)ist中會(huì)有兩段行程。我簡(jiǎn)單粗暴地把它拆成兩個(gè)變量,如section_a_list 和section_b_list。當(dāng)然,函數(shù)還是會(huì)返回一個(gè)名為flights_df 的DataFrame對(duì)象,有了它我們接下來(lái)就可以任意排序并視情況切片或合并。
變量名中帶a的表示第一段行程,帶b的表示第二段行程。接著看下一個(gè)函數(shù)。
別急,還有干貨!
到現(xiàn)在為止,我們有用于加載更多結(jié)果的函數(shù),有用于解析這些結(jié)果的函數(shù)。你可以認(rèn)為這就完事了,可以靠著它們?nèi)ナ謩?dòng)地爬網(wǎng)頁(yè)了,但我前面還提到過(guò),我們的目標(biāo)是能給自己發(fā)郵件,當(dāng)然還能包括一些其他信息??纯聪旅孢@個(gè)函數(shù)start_kayak,所有這些都在里面。
這需要我們定義一下要查詢的航班的地點(diǎn)和日期。我們會(huì)打開(kāi)kayak變量中的網(wǎng)址,并且查詢結(jié)果會(huì)直接按照“best”方式排序。在第一次爬數(shù)之后,我就獲得了頁(yè)面上方的價(jià)格矩陣數(shù)據(jù)集,它將用于計(jì)算均價(jià)和最低價(jià),然后和Kayak的預(yù)測(cè)價(jià)(頁(yè)面的左上角)一起通過(guò)電子郵件發(fā)出。在單個(gè)日期搜素時(shí)可能導(dǎo)致錯(cuò)誤,因?yàn)檫@種情況下頁(yè)面頂端沒(méi)有價(jià)格矩陣。
我用outlook郵箱(hotmail.com)做了測(cè)試。雖然Gmail我沒(méi)試過(guò),甚至還有其他各種郵箱,但我想應(yīng)該都沒(méi)問(wèn)題。而且我前文提到的書(shū)中也寫(xiě)了其他發(fā)郵件的方式,如果你有hotmail郵箱,可以直接在代碼中替換你的郵箱信息,就可以用了。
如果你想知道腳本中某部分代碼的功能,你要把那部分拷出來(lái)測(cè)試一下,因?yàn)橹挥羞@樣你才能徹底地理解它。
把代碼跑起來(lái)
當(dāng)然,我們還能把我們前面編的函數(shù)放進(jìn)循環(huán)里讓它一直執(zhí)行。寫(xiě)明4個(gè)輸入提示,包括起降的城市和起止時(shí)間(輸入)。但在測(cè)試的時(shí)候,我們并不想每次都去輸入這個(gè)四個(gè)變量,就直接修改4個(gè)變量,如注釋的那四行代碼所示。
恭喜各位,至此我們已經(jīng)大功告成了!其實(shí)還有很多可以改進(jìn)的地方,比如我能想到的還可以用Twilio實(shí)現(xiàn)發(fā)送短信進(jìn)而取代郵件。你還能架VPN或者以其他隱蔽的方式,同時(shí)通過(guò)多個(gè)服務(wù)器來(lái)爬數(shù)據(jù)。還有驗(yàn)證碼問(wèn)題,它們總會(huì)不時(shí)地跳出來(lái),不過(guò)這還是有辦法解決的。如果你有比較好的基礎(chǔ),我覺(jué)得你可以試試加上這些功能。甚至你還會(huì)想把Excel文件作為郵件的附件一起發(fā)出。
相關(guān)報(bào)道:
https://towardsdatascience.com/if-you-like-to-travel-let-python-help-you-scrape-the-best-fares-5a1f26213086
【本文是51CTO專(zhuān)欄機(jī)構(gòu)大數(shù)據(jù)文摘的原創(chuàng)文章,微信公眾號(hào)“大數(shù)據(jù)文摘( id: BigDataDigest)”】