please add plt.ion() between " plt.axis('off') " and " plt.show() "

[hawkingrei]

it can solve the problem that if you close the main windows,it will have a error