Построение трехмерного векторного поля с помощью quiver3d () в Mayavi - PullRequest
Новая
       37

Построение трехмерного векторного поля с помощью quiver3d () в Mayavi

0 голосов
/ 17 февраля 2020

Мне трудно понять, почему следующий код не работает внутри Jupyter Notebook! Любая помощь очень ценится 

%gui qt
import matplotlib.pyplot as plt
import numpy as np
from sympy import symbols
from mayavi import mlab

x,y,z = symbols('x y z')
def gradient(f):
    return (f.diff(x), f.diff(y),f.diff(z))

f = x*y**2+z**2
g = gradient(f)

xrange = np.linspace(-3,3,15)
yrange = np.linspace(-3,3,15)
zrange = np.linspace(-3,3,15)
X,Y,Z = np.meshgrid(xrange, yrange, zrange)

U = np.zeros((15,15,15))
V = np.zeros((15,15,15))
W = np.zeros((15,15,15))

for i in range(len(xrange)):
    for j in range(len(yrange)):
        for k in range(len(zrange)):
            x1 = X[i,j,k]
            y1 = Y[i,j,k]
            z1 = Z[i,j,k]
            U[i,j,k] = g[0].subs({x:x1, y:y1, z:z1})
            V[i,j,k] = g[1].subs({x:x1, y:y1, z:z1})
            W[i,j,k] = g[2].subs({x:x1, y:y1, z:z1})


mlab.quiver3d(X,Y,Z,U,V,W)

На самом деле, я не получаю никаких сообщений об ошибках, просто чёрное окно Mayavi. Я ждал более 10 минут, но перманенты windows черные. Я запускаю этот код на машине с Ubuntu 19.10. Для полноты картины хотелось бы отметить, что приведенный ниже код работает нормально. 

%gui qt
from mayavi import mlab
import numpy as np

def V(x, y, z):
    """ A 3D sinusoidal lattice with a parabolic confinement. """
    return np.cos(10*x) + np.cos(10*y) + np.cos(10*z) + 2*(x**2 + y**2 + z**2)
X, Y, Z = np.mgrid[-2:2:100j, -2:2:100j, -2:2:100j]
mlab.contour3d(X, Y, Z, V)

1 Ответ

0 голосов
/ 18 февраля 2020

Я мог бы решить эту проблему, добавив 

mlab.init_notebook()
mlab.clf()

после импорта модуля mlab . Таким образом, сюжет появляется внутри Jupyter Notebook.

...