oldj / pyheatmap

python heat map library

pyHeatMap

• Author: oldj
• Email: [email protected]
• Blog: https://oldj.net/
• Source: https://github.com/oldj/pyheatmap

pyHeatMap is a Python library for painting heat maps. It depends on Pillow. Python 2/3 compatible.

Screenshots

hit map

heat map

Install

by pip:

pip install pyheatmap

by easy_install:

easy_install pyheatmap

from source code:

git clone git://github.com/oldj/pyheatmap.git
cd pyheatmap
python setup.py install

Example:

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

import urllib
from pyheatmap.heatmap import HeatMap

def main():

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

    # start painting
    hm = HeatMap(data)
    hm.clickmap(save_as="hit.png")
    hm.heatmap(save_as="heat.png")

if __name__ == "__main__":
    main()

Copyright

This library is free and is provided under the MIT open source license.

Update

• 2015-08-31 Python 2/3 compatible.
• 2012-09-03 Create.

pyheatmap's Issues

Example运行时出错

提示说SystemError: tile cannot extend outside image

是否可以支持GPU加速运算

PIL支持GPU，热力图渲染可以利用GPU进行加速吗。

[bug]base backgound img. auto rotation? 背景图意外旋转?

~ 能否支持自定义的背景图

env.

• Darwin Kernel Version 16.7.0: Mon Nov 13 21:56:25 PST 2017
• Python 2.7.10
• download github mast repo., install by setup.py

err.

• 用 iPhone 7 拍摄背景图片
• 通过脚本完成 pyheatmap 的数据准备和输出调用
• 发现:
  • 如果是正方形照片的话, 最终输出时
    • 背景图片被意外向左旋转了 90度
  • 如果照片是长方形的, 输出的热力图吻合直觉

关键代码类似:
pyheatmap.heatmap(save_as='hm.png'
, r=36
, base= 'bgm.jpg')

suggest:

应该是 heatmap.py#L63
引发的意外...被随机使用了 w/h, 在相等的时候...

建议及时修正 ;-)

是否可以有图例

是否可以有热力图的图例，类似于什么颜色代表点密度是多少?

能否支持自定义的背景图

是否可以在一个图片基础上画热区图或点击图？

如果实现平滑模糊处理，有什么建议么？

类似heatmap.js的效果

有没有完整参数，能不能显示数值标签呢

关于float类型data

hi，现在有一个x,y均为float类型的二维data set，
根据您的code
def __heat(self, heat_data, x, y, n, template):
u""""""

    l = len(heat_data)
    width = self.width
    p = width * y + x

    for ip, iv in template:
        p2 = p + ip
        if 0 <= p2 < l:
            heat_data[p2] += iv * n

由于是float类型， heat_data[p2]报错list indices must be integer
我尝试修改源码，但并不太明白其中原理部分，有没有办法能支持float类型数据绘制heatmap？