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():
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], …] 这样的列表或元组,可以把它理解为一组平面坐标。
目前这个库可以生成两种图片:点击图、热图。
点击图效果如下:
热图效果如下:
绘制图片时,还可以指定一个底图,这个底图可以是任意图像,也可以是另一个点击图。比如:
|
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画圆算法。