plt.gca().set_aspect('equal', adjustable='box')
plt.gcf().canvas.mpl_disconnect(plt.gcf().canvas.manager.key_press_handler_id) plt.gcf().canvas.mpl_connect('key_press_event', onpress)
plt.figure(figsize=[10, 10]) plt.savefig('output.png')