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