pyHeatMap:使用Python绘制热图的库

pyHeatMap是一个使用Python生成热图的库,基本代码是我一年多之前写的,最近把它从项目中抠出来做成一个独立的库并开源。

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

pip install pyheatmap

# 或者

easy_install pyheatmap

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

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

import urllib
from pyheatmap.heatmap import HeatMap

def main():

    # 下载测试数据
    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="hit.png")
    hm.heatmap(save_as="heat.png")

if __name__ == "__main__":
    main()

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

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

点击图效果如下:

点击图

热图效果如下:

热图

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

def example2():

    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")

生成的图片形如:

关于绘制热图中用到的方法,可以参考我以前的文章,比如关于网页点击热区图关于热区图的色盘等,其中热图绘制中还用到了Bresenham画圆算法

这个库目前已经可以正常工作,不过也还有一些可优化和改进的地方。我会抽空继续完善它,如果你有兴趣参与,欢迎在GitHub上Fork它。

One Reply to “pyHeatMap:使用Python绘制热图的库”

发表评论

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / 更改 )

Twitter picture

You are commenting using your Twitter account. Log Out / 更改 )

Facebook photo

You are commenting using your Facebook account. Log Out / 更改 )

Google+ photo

You are commenting using your Google+ account. Log Out / 更改 )

Connecting to %s