本文主要是介绍Python应用指南：获取高德地铁站点数据（单城市版），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
书接上文，上篇文章是一次性下载全国所以城市的地铁站点数据，但是可视化的过程需要手动把换乘站给一个个复制出来分配到其他各个经过的线路，还需要核对站点顺序不能出错，如果只需要单个城市的数据呢？另外能不能直接生成shp图层？本篇文章就提出了优化方案，优化了脚本来抓取高德地图地铁站点数据，并给出excel或shp文件的下载方式以供选择。
获取单个城市的地铁数据，数据来源：高德地图 | 地铁图 (amap.com)
打开开发者模式快捷键Ctrl+Shift+I ,找到这个文件，双击打开；
我们就可以看到储存的地铁站点数据，包括城市编码、线路名称、线路编码、站点名称、站点编码、经度、纬度、POI编码、站点拼音等一系类标签，我们的目的就是获取这些信息，接下来直接放出完整代码，老规矩直接copy直接运行即可；
完整代码#运行环境Python 3.11
import requests
import geopandas as gpd
import pandas as pd
from datetime import datetimedef get_railway_stop(city_name, save_file=''):"""数据来源于高德地图: https://ditu.amap.com/:param city_name: 要爬取的地铁城市名，要正确输入:param save_file: 是否保存为文件1 表示存成 Excel 文件2 表示存成 Shapefile 文件"""def fetch_city_list():"""获取可以爬取地铁信息的城市列表:return: 城市列表的 JSON 数据或 None（如果请求失败）"""try:# 发送 GET 请求获取城市列表response = requests.get('https://map.amap.com/service/subway?_1707184339116&srhdata=citylist.json')response.raise_for_status() # 如果响应状态码不是 200，将抛出 HTTPError 异常return response.json()except requests.RequestException as e:# 打印错误信息print(f"Failed to fetch city list: {e}")return Nonedef fetch_subway_data(adcode, spell):"""根据城市编号和拼音获取地铁数据:param adcode: 城市编号:param spell: 城市拼音:return: 地铁数据的 JSON 数据或 None（如果请求失败）"""try:# 构造带有动态时间戳的 URLurl = f'https://map.amap.com/service/subway?&_={int(datetime.now().timestamp())}&srhdata={adcode}_drw_{spell}.json'# 发送 GET 请求获取地铁数据response = requests.get(url)response.raise_for_status() # 如果响应状态码不是 200，将抛出 HTTPError 异常return response.json()except requests.RequestException as e:# 打印错误信息print(f"Failed to fetch subway data: {e}")return None# 获取城市列表citylist = fetch_city_list()if citylist is None:# 如果获取城市列表失败，则返回 Nonereturn None# 查找指定城市的拼音和编号city_info = next((city for city in citylist['citylist'] if city['cityname'] == city_name), None)if not city_info:# 如果找不到指定城市，则输出错误信息并返回 Noneprint('城市名输入错误')return None# 获取该城市的拼音spell = city_info['spell']# 获取该城市的编号adcode = city_info['adcode']# 获取地铁数据subway_data = fetch_subway_data(adcode, spell)if subway_data is None:# 如果获取地铁数据失败，则返回 Nonereturn None# 获取文件名，例如 '深圳市地铁'filename = subway_data['s']# 初始化存储地铁站数据的列表station_data = []for line in subway_data['l']:# 遍历每条线路的地铁站for station in line['st']:# 将每个地铁站的信息添加到列表中station_data.append({'name': station['n'], # 地铁站名'line': f"{line['kn']} {line['la']}", # 地铁线路名'color': line['cl'], # 地铁线路颜色'poiid': station['poiid'], # 地铁线路id'lon': station['sl'].split(',')[0], # 经度'lat': station['sl'].split(',')[1], # 纬度'transfer': station['t'] # 是否换乘站 (1: 是)})# 创建 DataFramestations_df = pd.DataFrame(station_data)# 将经度和纬度列转换为数值类型stations_df[['lon', 'lat']] = stations_df[['lon', 'lat']].apply(pd.to_numeric)# 保存数据if save_file == 1:# 存成 Excel 文件stations_df.to_excel(f'{filename}.xlsx', index=False)print(f"Data saved to {filename}.xlsx")elif save_file == 2:# 存成 Shapefile 文件# 将 DataFrame 转换为 GeoDataFramegeo_stations_df = gpd.GeoDataFrame(stations_df,geometry=gpd.points_from_xy(stations_df['lon'], stations_df['lat']),crs='EPSG:4326')# 保存为 Shapefile 文件geo_stations_df.to_file(f'{filename}.shp', driver='ESRI Shapefile', encoding='utf-8')print(f"Data saved to {filename}.shp")# 返回 DataFramereturn stations_df# 上海市地铁站点坐标数据
get_railway_stop(city_name='上海市', save_file=1)
需要调整成其他城市，直接改下面的城市，另外这里提供了二种数据下载方式，也就是下面的1是Excel 格式，2是shp的点图层格式；
# 上海市地铁站点坐标数据
get_railway_stop(city_name='上海市', save_file=1)Excel 格式；
shp的点图层格式，同时直接检索【点集转线】，且线字段选择line即可连成直线，方法在上篇文章里详细说明过；
同样的由于我们拿到的数据是高德坐标系GCJ02，但我们通常的数据源可能是WGS84，所以我们需要转一下地理坐标系，关于如果点集转线和坐标系批量转换可以参考上篇文章，链接也放在这里：Python应用指南：获取高德地铁站点数据（全国版）-CSDN博客；
文章仅用于分享个人学习成果与个人存档之用，分享知识，如有侵权，请联系作者进行删除。所有信息均基于作者的个人理解和经验，不代表任何官方立场或权威解读。
这篇关于Python应用指南：获取高德地铁站点数据（单城市版）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
