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