python绘图 - pyheatmap基础案例

1.代码

from pyheatmap.heatmap import HeatMap

file_path = r'E:\yida\数据业务\12.txt'

fh = open(file_path, encoding='utf8', errors='ignore')

file_data = fh.readlines()

data = []

for line in file_data:

line = line.strip('\n')

line = line.split(',')

if len(line) != 2:

continue

line = [int(i) for i in line]

print(line)

 

hm = HeatMap(data)

hm.clickmap(save_as="d:/hit.png")

hm.heatmap(save_as="d:/heat.png")

 

2.结果图:

目前这个库可以生成两种图片：点击图、热图。

热图效果如下：

 

点击图效果如下：

 

标题

绘制图片时，还可以指定一个底图，这个底图可以是任意图像，也可以是另一个点击图。比如：

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