自定义场景、地图构建画热力图heatmap（python）

自定义场景、地图构建画热力图heatmap（python）

需求：给定一块区域地图，根据实验的数据拥堵量情况制作成热力图形式。并且，由于需要对比实验，两张热力图之间的热力值最大最小应该是一致的

本文链接：https://blog.csdn.net/weixin_42062224/article/details/121467466

实现过程用到的知识（可忽略）
1.bresenham画圆算法
根据近似和坐标关系递推获得像素格子位置

2.制作热区图色盘
https://oldj.net/article/2011/05/14/heat-map-colors/
http://oldj.net/article/hsl-to-rgb/

3.一些总结
原理：制作一张和底图一样大的热力图片覆盖在底图上，热力点作为画圆的中心点，圆的色块从中心到外围减弱，通过色盘颜色取。
实现参考：https://www.cnblogs.com/taotingz/p/11309333.html

直接使用方式

安装包

pip install ScenarioHeatMap==1.2.2

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的最大最小值