使用Python绘制热图的库 pyHea

发布时间:2019-08-31 09:55:36编辑:auto阅读(2230)

    pyHeatMap 详细介绍

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

    程序截图:

    点击图

    点击图

    热图

    热图

    安装:

    通过 pip 安装:

    pip install pyheatmap
    

    通过 easy_install 安装:

    easy_install pyheatmap
    

    通过源码安装:

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

    使用示例:

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

关键字