python快速绘图(三维图绘制)

Hua_Zhang

python快速绘图，简单绘制三维图（surface）

1. import numpy as np
   
2. import matplotlib.pyplot as plt
   
3. from mpl_toolkits.mplot3d import Axes3D
   
4. 
   
5. fig = plt.figure()
   
6. ax = Axes3D(fig)
   
7. 
   
8. X = np.arange(-4,4,0.25)
   
9. Y = np.arange(-4,4,0.25)
   
10. X,Y = np.meshgrid(X,Y)
    
11. R = np.sqrt(X**2+Y**2)
    
12. Z = np.cos(R)
    
13. ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))
    
14. ax.contourf(X,Y,Z,zdir='x',offset=-4,cmap=plt.get_cmap('rainbow'))
    
15. 
    
16. plt.savefig('test9.png',dpi=520)

复制代码

量子侠

谢谢楼主

开心耗子

谢谢楼主~

xxy

牛，这个挺好看的呢