需求:给定一块区域地图,根据需要制作成热力图形式。并且,由于需要对比实验,两张热力图之间的热力值最大最小应该是一致的
bresenham画圆算法
根据近似和坐标关系递推获得像素格子位置
制作热区图色盘
https://oldj.net/article/2011/05/14/heat-map-colors/
http://oldj.net/article/hsl-to-rgb/
安装包
pip install ScenarioHeatMap
from ScenarioHeatMap.SHM import * def demo_1(): """ # 无底图使用,无底图时必须图片给定长宽 :return: """ data = [[50, 50], [20, 20]] # 热力点坐标 hm = HeatMap(data, width=100, height=100) # width是宽即x轴 hm.data_transform([0, 0, 100, 100]) hm.heatmap() max_color, min_color = hm.get_max_min_heat() def demo_2(): """ 有底图使用,无需给定长宽 :return: """ # 底图 image = cv2.imread('map.jpg', cv2.IMREAD_UNCHANGED) # 热力数据点位置,二维列表的形式,对于我的数据而言热力点是经纬度 data = [[113.924888, 22.533647], [113.924901, 22.533148], ] # 创建heatmap对象 hm = HeatMap(data, base=image, colorful=True) # 给定底图左下角和右上角坐标 hm.data_transform([113.751647,22.40047, 114.622924, 22.855425]) # 绘制热力图,热力半径r默认10,自定义设置不能超过图片大小 hit_img = hm.heatmap(r=50) if __name__ == "__main__": demo_1() demo_2()
参数详解
1. HeatMap
data: 热力点数据
base: 给定底图
width: 图片的长
height: 图片的宽
screen_cross: 是否允许边界的色块在另一个方向出现。 默认False
transparency: 透明度。 默认0.5
colorful: 给定的地图是否彩色图片。 默认False
2.data_transform
postion:根据热力值的坐标而定,左下角坐标(Olng,Olat)右上角坐标(Dlng,Dlat)顺序组成的含有4个元素的list
3.heatmap
save_as:图片保存位置
r=10:热力半径
max_weight:给定热力权值最大值,默认为data数据计算所得的最大值
min_weight:给定热力权值最小值,默认为data数据计算所得的最小值
4.get_max_min_heat
获取当前热力data的最大最小值