python快速绘图(等值线图)

Hua_Zhang

python快速绘图，简单绘制二维等值线图，（类似matlab的contour与contourf,不过似乎更加好看）

1. import matplotlib.pyplot as plt
   
2. import numpy as np
   
3. 
   
4. def f(x,y):
   
5.     return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
   
6. 
   
7. n = 256
   
8. x = np.linspace(-3,3,n)
   
9. y = np.linspace(-3,3,n)
   
10. X,Y = np.meshgrid(x,y)
    
11. plt.axes([0.025,0.025,0.95,0.95])
    
12. plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)
    
13. C = plt.contour(X,Y,f(X,Y),8,color='black',linewidth=0.5)
    
14. plt.clabel(C,inline=1,fontsize=10)
    
15. plt.xticks([]),plt.yticks([])
    
16. # =============================================================================
    
17. # plt.show()
    
18. # =============================================================================
    
19. plt.savefig('test8.png',dpi=520)

复制代码

HOPE_WEST03

谢谢！已经试过了，效果很好！

Rainch

不错不错。