Попытка сделать контурную или цветную графику в 3D - PullRequest
0 голосов
/ 22 апреля 2019

У меня есть функция (I), которая определяет поле давления в контрольном объеме. Я создаю этот контрольный объем, объединяя 3 вектора (X, Y, Z) и применяю функцию I ко всем точкам сетки 3d-сетки.

Я пытаюсь найти способ построить каждую точку цветом, соответствующим величине, возвращаемой функцией I. Я чувствую, что это похоже на цветной график в 3d.

Возможно, я попытался сделать контурный график для каждого уровня в направлении z и наложить его на один трехмерный график

'' '

def ControlVolume(ress):
    x=np.linspace(-ress.X,ress.X,num=102);
    y=np.linspace(-ress.Y,ress.Y,num=102);
    z=np.linspace(-ress.Z,ress.Z,num=102);
    XV,YV,ZV=np.meshgrid(x,y,z)
    return XV,YV,ZV

def Isopotential(well,ress,x,y,z):
    I=(ress.mu/(4*np.pi*np.sqrt(ress.kx*ress.ky*ress.kz))*(well.q*
       ((x-ress.X)**2/ress.kx+(y-ress.Y)**2/ress.ky+(z-ress.Z)**2/ress.kz)
       **(-0.5)))
    return I

well1=well(10,10,10,5)
ress1=ress(20,20,20,10,10,1,0.3,0.2)
X,Y,Z=ControlVolume(ress1)
Ifield=Isopotential(well1,ress1,X,Y,Z)

'' '

В моей голове это должно выглядеть как поле, которое распространяется как концентрические сферы.

...