python抓取安居客小區(qū)數(shù)據(jù)
作者: 鄭曉 分類: Python 發(fā)布于: 2015-07-28 21:01 瀏覽:81,996 評論(8)
某功能需要一套城市所有小區(qū)的位置信息數(shù)據(jù),一開始是使用的百度地圖api來進行關(guān)鍵詞搜索,勉強能用,但數(shù)據(jù)量非常少,還是有大量的社區(qū)/小區(qū)搜不到。
周末在家上網(wǎng)時發(fā)現(xiàn)安居客上直接就有每個城市的小區(qū)大全,欣喜若狂,于是就立即寫了個爬蟲試試。
以下貼代碼,python2.7,lxml+request庫。
#coding=utf-8
#author : zx
#date : 2015/07/27
import requests
import MySQLdb
import time
import string
import random
from lxml import etree
#ua頭信息 get時可以隨機使用
headers = [
{ "User-Agent":"Mozilla/5.0 (Linux; U; Android 4.1; en-us; GT-N7100 Build/JRO03C) AppleWebKit/534.30 (KHTML, like Gecko) Version/4.0 Mobile Safari/534.30"},
{ "User-Agent":"Mozilla/5.0 (compatible; MSIE 10.0; Windows Phone 8.0; Trident/6.0; IEMobile/10.0; ARM; Touch; NOKIA; Lumia 520)"},
{ "User-Agent":"Mozilla/5.0 (BB10; Touch) AppleWebKit/537.10+ (KHTML, like Gecko) Version/10.0.9.2372 Mobile Safari/537.10+"},
{ "User-Agent":"Mozilla/5.0 (Linux; Android 4.4.2; GT-I9505 Build/JDQ39) AppleWebKit/537.36 (KHTML, like Gecko) Version/1.5 Chrome/28.0.1500.94 Mobile Safari/537.36"}
]
#城市入口頁面
#我只抓的青島本地
#其它城市或全國城市可通過這個頁面抓取城市列表http://m.anjuke.com/cityList
url = 'http://m.anjuke.com/qd/xiaoqu/'
req = requests.get(url)
cookie = req.cookies.get_dict()
#鏈接數(shù)據(jù)庫
conn = MySQLdb.connect('localhost', '*****', '******', '***', charset='utf8')
cursor = conn.cursor()
sql = "insert into xiaoqu (name, lat, lng, address, district) values (%s, %s, %s, %s, %s)"
sql_v = []
page = etree.HTML(req.text)
districtHTML = page.xpath(u"http://div[@class='listcont cont_hei']")[0]
#采集目標城市的各行政區(qū)域url
#當然如果不想?yún)^(qū)分行政區(qū)可以直接抓“全部” 即上面url中的所有小區(qū)及分頁
districtUrl = {}
i = 0
for a in districtHTML:
if i==0:
i = 1
continue
districtUrl[a.text] = a.get('href')
#開始采集
total_all = 0
for k,u in districtUrl.items():
p = 1 #分頁
while True:
header_i = random.randint(0, len(headers)-1)
url_p = u.rstrip('/') + '-p' + str(p)
r = requests.get(url_p, cookies=cookie, headers=headers[header_i])
page = etree.HTML(r.text) #這里轉(zhuǎn)換大小寫要按情況...
communitysUrlDiv = page.xpath(u"http://div[@class='items']")[0]
total = len(communitysUrlDiv)
i = 0
for a in communitysUrlDiv:
i+=1
r = requests.get(a.get('href'), cookies=cookie, headers=headers[header_i])
#抓取時發(fā)現(xiàn)有少量404頁會直接導致程序報錯退出- -!
#唉 說明代碼寫的還不夠健壯啊
#加了if判斷和try, 錯誤時可以跳過或做一些簡單處理和調(diào)試...
if r.status_code == 404:
continue
page = etree.HTML(r.text)
try:
name = page.xpath(u"http://h1[@class='f1']")[0].text
except:
print a.get('href')
print r.text
raw_input()
#有少量小區(qū)未設(shè)置經(jīng)緯度信息
#只能得到它的地址了
try:
latlng = page.xpath(u"http://a[@class='comm_map']")[0]
lat = latlng.get('lat')
lng = latlng.get('lng')
address = latlng.get('address')
except:
lat = ''
lng = ''
address = page.xpath(u"http://span[@class='rightArea']/em")[0].text
sql_v.append((name, lat, lng, address, k))
print "\r\r\r",
print u"正在下載 %s 的數(shù)據(jù),第 %d 頁,共 %d 條,當前:".encode('gbk') %(k.encode('gbk'),p, total) + string.rjust(str(i),3).encode('gbk'),
time.sleep(0.5) #每次抓取停頓
#執(zhí)行插入數(shù)據(jù)庫
cursor.executemany(sql, sql_v)
sql_v = []
time.sleep(5) #每頁完成后停頓
total_all += total
print ''
print u"成功入庫 %d 條數(shù)據(jù),總數(shù) %d".encode('gbk') % (total, total_all)
if total < 500: break else: p += 1#及時關(guān)閉數(shù)據(jù)庫 做個好孩子 任務(wù)完成~cursor.close()conn.close()print u'所有數(shù)據(jù)采集完成! 共 %d 條數(shù)據(jù)'.encode('gbk') % (total_all)raw_input()
注釋我覺得已經(jīng)寫的很詳細了,在cmd中顯示,字符串當然要轉(zhuǎn)一下碼。
以下是運行狀態(tài)和得到的數(shù)據(jù)截圖。
本文采用知識共享署名-非商業(yè)性使用 3.0 中國大陸許可協(xié)議進行許可,轉(zhuǎn)載時請注明出處及相應(yīng)鏈接。
本文永久鏈接: http://yjfs.org.cn/python-xiaoqu-data.html
加我微信8344290