Question

我很难弄清楚为什么以下代码在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分钟以上，但窗户永久物变黑了。我在装有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)

Answer 1

我可以通过添加来解决问题

mlab.init_notebook()
mlab.clf()

导入 mlab 模块后，

。这样，情节就会出现在Jupyter Notebook内部。

在Mayavi中使用quiver3d（）绘制3D矢量场

1 个答案: