使用Python绘制热图的库 pyHea

2020-01-09 12:33:19 浏览数 (1)

pyHeatMap 详细介绍

这是一个生成热图的小程序,基于 Python 和 PIL 开发。

程序截图:

点击图

热图

安装:

通过 pip 安装:

代码语言:javascript复制
pip install pyheatmap

通过 easy_install 安装:

代码语言:javascript复制
easy_install pyheatmap

通过源码安装:

代码语言:javascript复制
git clone git://github.com/oldj/pyheatmap.git
cd pyheatmap
python setup.py install

使用示例:

代码语言:javascript复制
# -*- 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()

0 人点赞