python 获取百度地图上北京地铁的数据,然后excel导出

今天有人叫我写一下怎样获取百度上地铁的站的名字和线路的代码,我顺手写了一下,这里分享一下我的过程:

import requests
import json

import pandas as pd


# url='https://map.baidu.com/?qt=subways&c=131&format=json&t=1589545402711&callback=jsonp96324632'
url='https://map.baidu.com/?qt=subways&c=131&format=json&t=1589545402711'
res = requests.get(url=url)
print(res.text) 
json_data=json.loads(res.text)

 
with open('./data.json', 'w') as json_file:
    json_file.write(res.text)



def getdata(subway):
    list_data=[]
    for item in subway['l']:
        for stop in item['p']:
            if('uid' in stop['p_xmlattr']):
                uid=stop['p_xmlattr']['uid']
            else:
                uid=0
                print(stop['p_xmlattr'])
    #         print(uid)
            sid=stop['p_xmlattr']['sid']
            
            if(sid==''):
                continue
            px=stop['p_xmlattr']['px']
            py=stop['p_xmlattr']['py']
            ln=stop['p_xmlattr']['ln']
       
            if('|' in ln):
                line=ln
                city='北京市'
            else:
                line=ln
                city='北京市'
            
            prod=[uid,city,line,sid,px,py]
            list_data.append(prod)
    return list_data


subway=json_data['subways']
list_data=getdata(subway)
column_name = ['id','城市','地铁线', '站名','px','py']
csv_name='data.xls'
xml_df = pd.DataFrame(list_data, columns=column_name)
xml_df.to_excel(csv_name, index=None) 

猜你喜欢

转载自blog.csdn.net/w5688414/article/details/106151031