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