豆粑粑 mayavi 3d contour surface python

meatball1982

画3D contour surface，matlab，暂时是实现不了。
用python 下的mayavi2实现的，如何安装完全按官网的教程。

运行一个例子，小华丽。

1. from scipy.interpolate import griddata
   
2. import numpy as np
   
3. 
   
4. # Create some test data, 3D gaussian, 200 points
   
5. dx, pts = 2, 50j
   
6. 
   
7. N = 500
   
8. R = np.random.random((N,3))*2*dx - dx
   
9. V = np.exp(-( (np.sin(R)**2).sum(axis=1)) )
   
10. 
    
11. print "Rsize=",np.shape(R)
    
12. print "R=" ,R
    
13. print "V=" ,V
    
14. 
    
15. 
    
16. # Create the grid to interpolate on
    
17. X,Y,Z = np.mgrid[-dx:dx:pts, -dx:dx:pts, -dx:dx:pts]
    
18. 
    
19. # Interpolate the data
    
20. F = griddata(R, V, (X,Y,Z))
    
21. #np.savetxt('./F.txt',F)
    
22. print "F size", np.size(F)
    
23. print "grid val= ",F
    
24. 
    
25. from mayavi import mlab
    
26. mlab.contour3d(F,contours=8,opacity=.2 )
    

复制代码