python基础编程：python pyheatmap包绘制热力图

这篇文章主要为大家详细介绍了python pyheatmap包绘制热力图，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
利用python pyheatmap包绘制热力图，供大家参考，具体内容如下

import matplotlib.pyplot as plt
from pyheatmap.heatmap import HeatMap
 
def plot_data(filename):
 with open(filename,'r') as fh:
  data=fh.read().split('\n')
 xs = []
 ys = []
 data_test=[]
 for line in data:
  line=line.strip().split()
  if len(line)>3:
   opt, x, y = line[0], line[1], line[2]
   if opt == '0':
    xs.append(int(x))
    ys.append(int(y))
    data_test.append([int(x),int(y)])
 
 plt.xlim()
 plt.ylim()
 plt.xlabel("x")
 plt.ylabel("y")
 plt.plot(xs, ys, 'ro')
 plt.show()
 return data_test
 
 
filename='track.log'
data=plot_data(filename) 
 
# 开始绘制
hm = HeatMap(data)
hm.clickmap(save_as="hit.png")
hm.heatmap(save_as="heat.png")
 
# 绘制带背景的点击热图
hm2 = HeatMap(data)
hit_img2 = hm2.clickmap(ba