python The pyheatmap package draws the heat map

  • 2021-01-25 07:46:42
  • OfStack

The python pyheatmap package is used to draw the heat map for your reference. The specific content is as follows


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) 

#  Began to draw 
hm = HeatMap(data)
hm.clickmap(save_as="hit.png")
hm.heatmap(save_as="heat.png")

#  Draw a click heat map with a background 
hm2 = HeatMap(data)
hit_img2 = hm2.clickmap(base='base.png') # base.png Is the background image 
hit_img2.save("hit2.png")

Get the mouse position


import time
import pyautogui as pag


while True:
 #print("Press Ctrl-C to end")
 screenWidth, screenHeight = pag.size() # Gets the size of the screen 
 #print(screenWidth,screenHeight)
 x,y = pag.position() # Gets the current mouse position 
 print(x,y)
 time.sleep(0.1)


Read the mouse click position


import pythoncom, pyHook
def onMouseEvent(event):
  print("Position:", event.Position)
  return True
def main():
 hm = pyHook.HookManager()
 hm.HookKeyboard()
 hm.MouseAllButtonsDown = onMouseEvent
 hm.MouseAllButtonsUp = onMouseEvent
 hm.HookMouse()
 pythoncom.PumpMessages()
if __name__ == "__main__":
 main()

Related articles: