如何通过Python编写程序利用百度地图API显示地图实时路况？

pip install requests
pip install folium

import requests
import folium

# 获取百度地图API的密钥
api_key = 'your_api_key'

# 获取实时路况信息
def get_traffic_info(api_key):
    url = f'http://api.map.baidu.com/traffic/v1/road?ak={api_key}&city=北京市'
    response = requests.get(url)
    traffic_info = response.json()
    return traffic_info

# 在地图上显示实时路况
def show_traffic_on_map(traffic_info):
    # 创建地图对象
    map = folium.Map(location=[39.9075, 116.39723], control_scale=True, zoom_start=11)
    
    # 遍历实时路况信息
    for road in traffic_info['roads']:
        polyline = road['polyline']
        status = road['status']
        
        # 根据路况状态选择不同的颜色
        if status == 1:
            color = 'green'
        elif status == 2:
            color = 'yellow'
        elif status == 3:
            color = 'red'
        else:
            color = 'gray'
        
        # 在地图上添加路线
        folium.PolyLine(locations=polyline, color=color, weight=5).add_to(map)
    
    # 保存地图为HTML文件，可在浏览器中打开查看
    map.save('traffic_map.html')
    print('地图已保存为 traffic_map.html')

# 主函数
def main(api_key):
    traffic_info = get_traffic_info(api_key)
    show_traffic_on_map(traffic_info)

if __name__ == '__main__':
    main(api_key)