Python如何爬取高德地圖地鐵線路及站點數據

這篇文章給大家分享的是有關Python如何爬取高德地圖地鐵線路及站點數據的內容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。

前言

本文的文字及圖片來源于網絡,僅供學習、交流使用,不具有任何商業用途,版權歸原作者所有,如有問題請及時聯系我們以作處理。

# coding=utf-8

import requests
import time
import json
import ast
import os
import utils
from lxml import etree

PAGE_URL = 'http://map.amap.com/subway/index.html?&1100'
DATA_URL = 'http://map.amap.com/service/subway?srhdata='
HEADER = {
    "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/71.0.3578.98 Safari/537.36"}


def fetchAllCity(url, header):
    r = requests.get(url, header)
    html = r.content
    element = etree.HTML(html)
    options = element.xpath("//a[contains(@class, 'city')]")

    cities = []
    for option in options:
        city = {
            'id': option.get('id'),
            'name': option.get('cityname'),
            'text': option.text
        }
        cities.append(city)
    return cities



def parseCityData(citys):
    lw = open('./lwkt.txt', 'w')
    lw.write('wkt' + '\n')
    pw = open('./pwkt.txt', 'w')
    pw.write('wkt' + '\n')
    for city in citys:
        parseCityPointFromApi(city, lw, pw)


def parseCityPointFromApi(city, lw, pw):
    url = DATA_URL + "{}_drw_{}.json".format(city['id'], city['name'])
    print(url)
    json_str = requests.get(url).text
    res = json.loads(json_str)
    res = res['l']
    for r in res:
        st = r['st']
        coords = []
        for s in st:
            _coords = s.get('sl', '').split(',')
            coords.append(_coords)
            pwkt = '"POINT ({} {})"'.format(_coords[0], _coords[1])
            pw.write(pwkt + '\n')
        new_coords = ','.join(['%s %s' % (coord[0], coord[1]) for coord in coords])
        lwkt = '"LINESTRING(%s)"' % new_coords
        lw.write(lwkt + '\n')



def main():
    cities = fetchAllCity(PAGE_URL, HEADER)
    print(cities)
    parseCityData(cities)



if __name__ == '__main__':
    main()

感謝各位的閱讀！關于“Python如何爬取高德地圖地鐵線路及站點數據”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，讓大家可以學到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！