Mayavi结合了两个隐式三维表面

时间:2016-11-07 08:48:41

标签: python plot 3d mayavi

我使用Mayavi绘制隐式3D表面,我想将这些表面组合成一个图。但是,当我这样做时,我得到类似的东西:

enter image description here

我正在使用的代码:

import numpy as np
from mayavi import mlab

A=0.24639243776
B=5.39100472027e-17
C=1.71555149594
D=1.72967325617
E=7.50535440036
F=-1.17072847143
G=1.0

x, y, z = np.ogrid[-3:1:100j, -10:1:100j, 0:3:100j]

def Fun (x,y,z, A,B,C,D,E,F,G):
    F1 = (x - A + y - B) / 2 + np.sqrt(((x - A - y + B) / 2) ** 2 + C * z ** 2)
    F2 = np.sqrt(x*((F*y)/2 + (E*x)/D) + y*((F*x)/2 + (D*y)/E) + G*z**2) - np.sqrt(D*E)
    F1 [F1 < 0] = F2 [F1 < 0]
    return F1

Fu = Fun (x,y,z,A,B,C,D,E,F,G)

mlab.contour3d(Fu, contours = [0])
mlab.show()

问题是无论我怎么做,其中一个表面的聚合物非常低!我在Mathematica中绘制了相同的图,它产生了一个完美的情节:

enter image description here

我不希望得到与Mathematica是商业产品相同的质量图。但至少我希望两个表面都是光滑的。

如果Mayavi不是正确的任务工具,也许你可以推荐一些我可以达到预期结果的东西。

提前致谢!

1 个答案:

答案 0 :(得分:0)

我明白了。关键是将曲面设置为“无”以进行修剪。然后可以以相同的方式绘制另一个表面。此外,对于根据比例绘制的表面,x,y,z也必须传递到contour3d

import numpy as np
from mayavi import mlab

A=0.24639243776
B=5.39100472027e-17
C=1.71555149594
D=1.72967325617
E=7.50535440036
F=-1.17072847143
G=1.0

x, y, z = np.mgrid[-3:1:300j, -10:1:300j, 0:3:300j]

def Fun1 (x,y,z, A,B,C,D,E,F,G):
    F1 = (x - A + y - B) / 2 + np.sqrt(((x - A - y + B) / 2) ** 2 + C * z ** 2)
    F2 = np.sqrt(x*((F*y)/2 + (E*x)/D) + y*((F*x)/2 + (D*y)/E) + G*z**2) - np.sqrt(D*E)
    F1 [F2 > 0.0] = None
    return F1


def Fun2 (x,y,z, A,B,C,D,E,F,G):
    F1 = (x - A + y - B) / 2 + np.sqrt(((x - A - y + B) / 2) ** 2 + C * z ** 2)
    F2 = np.sqrt(x*((F*y)/2 + (E*x)/D) + y*((F*x)/2 + (D*y)/E) + G*z**2) - np.sqrt(D*E)
    F2 [F1 > 0.0] = None
    return F2

Fu1 = Fun1 (x,y,z,A,B,C,D,E,F,G)
Fu2 = Fun2 (x,y,z,A,B,C,D,E,F,G)

mlab.contour3d(x,y,z,Fu1, contours = [0])
mlab.contour3d(x,y,z,Fu2, contours = [0])
mlab.show()

结果如下:

enter image description here