Python教程

自定义场景化热力图(python)

本文主要是介绍自定义场景化热力图(python),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

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

在这里插入图片描述

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

这篇关于自定义场景化热力图(python)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!