pyHeatMap：使用Python绘制热图的库

pyHeatMap是一个使用Python生成热图的库，基本代码是我一年多之前写的，最近把它从项目中抠出来做成一个独立的库并开源。(https://github.com/oldj/pyheatmap)

可以直接下载源码安装最新的版本，也可以通过pip或easy_install安装稳定的发布版：

1 2 3 4 5

pip install pyheatmap   # 或者   easy_install pyheatmap

pyHeatMap依赖于PIL库，使用之前需要先确保你的环境中已经安装了PIL。使用方法很简单，比如：

# -*-coding:utf-8-*-

import urllibfrom pyheatmap.heatmap 
import HeatMap
def main():    
data = []    # download test data    url="https://raw.github.com/oldj/pyheatmap/master/examples/test_data.txt"    sdata = urllib.urlopen(url).read().split("\n")
    data = []
    for ln in sdata:
        a = ln.split(",")
            if len(a) != 2:
                        continue        a = [int(i) 
                        for i in a]
        data.append(a)

    hm = HeatMap(data)
    hm.clickmap(save_as="d://python/hit.png")
    hm.heatmap(save_as="d://python/heat.png")
    if __name__ == "__main__":
    main()

输入的数据为形如 [[123, 234], [429, 82], [220, 535], …] 这样的列表或元组，可以把它理解为一组平面坐标。

目前这个库可以生成两种图片：点击图、热图。

点击图效果如下：

热图效果如下：

绘制图片时，还可以指定一个底图，这个底图可以是任意图像，也可以是另一个点击图。比如：

1 2 3 4 5 6 7 8 9 10

defexample2():       data_1=loadDataFromFile("test_data.txt")     data_2=loadDataFromFile("test_data2.txt")       hm=HeatMap(data_1)     hit_img=hm.clickmap()     hm2=HeatMap(data_2)     hit_img2=hm2.clickmap(base=hit_img,color=(0,0,255,255))     hit_img2.save("hit2.png")

生成的图片形如：

关于绘制热图中用到的方法，可以参考我以前的文章，比如

关于网页点击热区图、

http://oldj.net/article/page-heat-map/

关于热区图的色盘

http://oldj.net/article/heat-map-colors/

其中热图绘制中还用到了

Bresenham画圆算法

http://oldj.net/article/bresenham-algorithm/