专栏名称: CSDN
CSDN精彩内容每日推荐。我们关注IT产品研发背后的那些人、技术和故事。
51好读  ›  专栏  ›  CSDN

国庆去哪里人最多?用 Python 抓取的热力图告诉你!

CSDN  · 公众号  · 科技媒体  · 2017-09-30 12:28

正文

点击上方“ CSDN ”,选择“置顶公众号”

关键时刻,第一时间送达!


国庆倒计时1天,我们即将迎来8天的小长假,相信很多小伙伴们已经提前规划国庆去哪儿?你是选择去人最少的单位加班呢?还是选择人山人海的景点观光?


如果去人最少的单位加班,一定要记得告知老板,以望升职加薪走上人生巅峰;


如果选择人山人海的景点,一定要提前查看攻略,比如下文中使用 Python 技术抓取热门景点的热力图。


金秋九月,丹桂飘香,在这秋高气爽,阳光灿烂的收获季节里,我们送走了一个个暑假余额耗尽哭着走向校园的孩子们,又即将迎来一年一度伟大祖国母亲的生日趴体(无心上班,迫不及待想为祖国母亲庆生)。



那么问题来了,去哪儿玩呢?


百度输了个“国庆”,出来的第一条居然是“国庆去哪里旅游人少”。



于是我想到可否通过旅游网站的景点销量来判断近期各景点流量情况呢? 本文按去哪儿网景点页面,得到景点的信息。


1.百度的地图API和echarts


本次主要抓取的内容是数据,因此我决定用图表来输出抓取的数据,也就意味着我要用爬取的景点销量以及景点的具体位置来生成一些可视化数据。


在这里使用的是百度的地图API和echarts,前者是专门提供地图API的工具,后者是数据处理居家旅行的好伙伴。


那么API是什么?API是应用程序的编程接口,就好像插头与插座一样,我们的程序需要电,插座中提供了电,我们只需要在程序中写一个与插座匹配的插头接口,就可以使用电来做我们想做的事情,而不需要知道电是如何产生的。


引入数据后的百度热力图


而具体流程如下,好比出书的过程,想要将完结的小说出书,首先要通过出版社的出版服务,提供正文内容和封面设计图,将保存好的 word 文本和封面图发给出版社,由此得到一本书。在此过程中,并不需要了解出版社印刷、装订的过程。


通过API对接的开发者与服务商


2.确定输出文件


百度地图提供了很多API使用示例,有HTML基础,有JS基础就可以尝试改函数了。仔细观察源代码,可以知道热力图的生成主要的数据都存放在points这个变量中。



这种[{x:x,x:x},{x:x,x:x}]格式的数据,是一种json格式的数据,由于具有自我描述性,所以比较通俗易懂,大概可以知道这里的三个值,前俩个是经纬度,最后一个应该是权重。


也就是说,如果我希望将景点的热门程度生成为热力图,我需要得到景点的经纬度,以及它的权重,景点的销量可以作为权重,并且这个数据应该是json格式的呈现方式。 echarts也是一样的。


3.爬取数据


其实这次的爬虫部分是比较简单的。


分析网址(去哪儿景点)→爬取分页中信息(景点经纬度、销量)→转为json文件。


分析去哪儿景点页的网址可得出结构:


这次没有用正则来匹配内容,而使用了xpath匹配。


def getList():

place = raw_input('请输入想搜索的区域、类型(如北京、热门景点等):')

url = 'http://piao.qunar.com/ticket/list.htm?keyword='+ str(place)


def getList():

place = raw_input('请输入想搜索的区域、类型(如北京、热门景点等):')

url = 'http://piao.qunar.com/ticket/list.htm?keyword='+ str(place) +'®ion=&from=mpl_search_suggest&page={}'

i = 1

sightlist = []

while i:

page = getPage(url.format(i))

selector = etree.HTML(page)

print '正在爬取第' + str(i) + '页景点信息'

i+=1

informations = selector.xpath('//div[@class="result_list"]/div')

for inf in informations: #获取必要信息

sight_name = inf.xpath('./div/div/h3/a/text()')[0]

sight_level = inf.xpath('.//span[@class="level"]/text()')

if len(sight_level):

sight_level = sight_level[0].replace('景区','')

else:

sight_level = 0

sight_area = inf.xpath('.//span[@class="area"]/a/text()')[0]

sight_hot = inf.xpath('.//span[@class="product_star_level"]//span/text()')[0].replace('热度 ','')

sight_add = inf.xpath('.//p[@class="address color999"]/span/text()')[0]

sight_add = re.sub('地址:|(.*?)|\(.*?\)|,.*?$|\/.*?$','',str(sight_add))

sight_slogen = inf.xpath('.//div[@class="intro color999"]/text()')[0]

sight_price = inf.xpath('.//span[@class="sight_item_price"]/em/text()')

if len(sight_price):

sight_price = sight_price[0]

else:

i = 0

break

sight_soldnum = inf.xpath('.//span[@class="hot_num"]/text()')[0]

sight_url = inf.xpath('.//h3/a[@class="name"]/@href')[0]

sightlist.append([sight_name,sight_level,sight_area,float(sight_price),int(sight_soldnum),float(sight_hot),sight_add.replace('地址:',''),sight_slogen,sight_url])

time.sleep(3)

return sightlist,place


  1. 这里把每个景点的所有信息都抓取下来。

  2. 使用了while循环,for循环的break的方式是发现无销量时给i值赋零,这样while循环也会同时结束。

  3. 地址的匹配使用re.sub()函数去除了n多复杂信息,这点后面解释。


4.输出本地文本


为了防止代码运行错误,为了维护代码运行的和平,将输出的信息列表存入到excel文件中了,方便日后查阅,很简单的代码,需要了解pandas的用法。


def listToExcel(list,name):

df = pd.DataFrame(list,columns=['景点名称','级别','所在区域','起步价','销售量','热度','地址','标语','详情网址'])

df.to_excel(name + '景点信息.xlsx')


5.百度经纬度API


百度经纬度API网址:http://api.map.baidu.com/geocoder/v2/?address=地址&output=json&ak=百度密钥,修改网址里的“地址”和“百度密钥”,在浏览器打开,就可以看到经纬度的json信息。


#上海市东方明珠的经纬度信息

{"status":0,"result":{"location":{"lng":121.5064701060957,"lat":31.245341811634675},"precise":1,"confidence":70,"level":"UNKNOWN"}}


这样我就可以根据爬到的景点地址,查到对应的经纬度!python获取经纬度json数据的代码如下。


def getBaiduGeo(sightlist,name):

ak = '密钥'

headers = {

'User-Agent' :'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36'

}

address = 地址

url = 'http://api.map.baidu.com/geocoder/v2/?address=' + address  + '&output=json&ak=' + ak

json_data = requests.get(url = url).json()

json_geo = json_data['result']['location']


观察获取的json文件,location中的数据和百度api所需要的json格式基本是一样,还需要将景点销量加入到json文件中,这里可以了解一下json的浅拷贝和深拷贝知识,最后将整理好的json文件输出到本地文件中。


def getBaiduGeo(sightlist,name):

ak = '密钥'

headers = {

'User-Agent' :'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36'

}

list = sightlist

bjsonlist = []

ejsonlist1 = []

ejsonlist2 = []

num = 1

for l in list:

try:

try:

try:

address = l[6]

url = 'http://api.map.baidu.com/geocoder/v2/?address=' + address  + '&output=json&ak=' + ak

json_data = requests.get(url = url).json()

json_geo = json_data['result']['location']

except KeyError,e:

address = l[0]

url = 'http://api.map.baidu.com/geocoder/v2/?address=' + address  + '&output=json&ak=' + ak

json_data = requests.get(url = url).json()

json_geo = json_data['result']['location']

except KeyError,e:

address = l[2]

url = 'http://api.map.baidu.com/geocoder/v2/?address=' + address  + '&output=json&ak=' + ak

json_data = requests.get(url = url).json()

json_geo = json_data['result']['location']

except KeyError,e:

continue

json_geo['count'] = l[4]/100

bjsonlist.append(json_geo)

ejson1 = {l[0] : [json_geo['lng'],json_geo['lat']]}

ejsonlist1 = dict(ejsonlist1,**ejson1)

ejson2 = {'name' : l[0],'value' : l[4]/100}

ejsonlist2.append(ejson2)

print '正在生成第' + str(num) + '个景点的经纬度'

num +=1

bjsonlist =json.dumps(bjsonlist)

ejsonlist1 = json.dumps(ejsonlist1,ensure_ascii=False)

ejsonlist2 = json.dumps(ejsonlist2,ensure_ascii=False)

with open('./points.json',"w") as f:

f.write(bjsonlist)

with open('./geoCoordMap.json',"w") as f:

f.write(ejsonlist1)

with open('./data.json',"w") as f:

f.write(ejsonlist2)


在设置获取经纬度的地址时,为了匹配到更准确的经纬度,我选择了匹配景点地址,然而,景点地址里有各种神奇的地址,因此有了第三部门中去除复杂的信息。


然而,就算去掉了复杂信息,还有一些匹配不到的景点地址,于是我使用了嵌套try,如果景点地址匹配不到;就匹配景点名称,如果景点名称匹配不到;就匹配景点所在区域,如果依然匹配不到,那就跳过。


这里生成的三个json文件,一个是给百度地图api引入用的,另俩个是给echarts引入用的。


6.网页读取json文件


将第二章中所述的百度地图api示例中的源代码复制到解释器中,添加密钥,保存为html文件,打开就可以看到和官网上一样的显示效果。echarts需要在实例页面,点击页面右上角的EN切换到英文版,然后点击download demo下载完整源代码。


根据html导入json文件修改网页源码,导入json文件。


#百度地图api示例代码中各位置修改部分







请到「今天看啥」查看全文