Python生成地理位置热力图

# author：zhangpengxu

import numpy as np
import pandas as pd
import folium
import webbrowser
from folium.plugins import HeatMap
posi=pd.read_excel('/Users/zhangpengxu/Desktop/text.xlsx')
num = 142
lat = np.array(posi["PickLatitude"][0:num])          # PickLatitude为Excel的列获取维度之维度值
lon = np.array(posi["PickLongitude"][0:num])        # 获取经度值
# pop = np.array(posi["POP"][0:num],dtype=float)    # 获取人口数，转化为numpy浮点型
# gdp = np.array(posi["GDP"][0:num],dtype=float)    # 获取GDP数，转化为numpy浮点型
gdp_average = 1
#np.array(posi["GDP_Average"][0:num],dtype=float) # 获取人均GDP数，转化为numpy浮点型
#
data = [ [lat[i],lon[i]] for i in range(num) ]    #将数据制作成[lats,lons,weights]的形式
map_osm = folium.Map(location=[1,2],zoom_start=6)    #绘制Map，开始缩放程度是5倍
HeatMap(data).add_to(map_osm)  # 将热力图添加到前面建立的map里

file_path = r"/Users/zhangpengxu/Desktop/text.html"
map_osm.save(file_path)     # 保存为html文件

webbrowser.open(file_path)  # 默认浏览器打开