Question

我正在尝试使用plot_wireframe模块中的Axes3D绘制线框。我的数据集：三个2D数组：两个用np.arange然后np.meshgrid创建。第三个是包含函数“f2”（f2 = f（x，y））的结果的数组。我期待3D绘图，但结果是3D空间中的2D绘图。现在我制作了一些代码，看起来像是mpl的文档化的一部分，但仍然不起作用。代码：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

def f2(x, y):
    '''Return f(x, y) = exp(-(x * x + y * y)) * sin(-5. * (x * x + y * y))'''    
    return np.exp(-(x**2 + y**2)) * np.sin(-5.0 * (x**2 + y**2))


fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x2 = np.arange(-1.5, 1.5, 0.02)
y2 = np.arange(1.5, -1.5, -0.02)

X, Y = np.meshgrid(x2, y2)

z2 = f2(X, Y)

ax.plot_wireframe(X, X, z2, rstride=10, cstride=10, linewidth=0.1,
                     label='$\exp(-(x^2 + y^2)){sin}(-5.0(x^2 + y^2)$')

plt.show()

这就是我想要的： Beautiful 3D plot 但现实是残酷的： "A plot I've got"

我不知道我做错了什么。这是向量值的问题吗？

Answer 1

更改

ax.plot_wireframe(X, X, z2, rstride=10, cstride=10, linewidth=0.1,
                 label='$\exp(-(x^2 + y^2)){sin}(-5.0(x^2 + y^2)$')

到

ax.plot_wireframe(X, Y, z2, rstride=10, cstride=10, linewidth=0.1,
                 label='$\exp(-(x^2 + y^2)){sin}(-5.0(x^2 + y^2)$')

您使用X代替Y作为ax.plot_wireframe的第二个参数。因此，所有点都落在对角线Y=X之上。

matplotlib 1.5中的Wire_frame无法正确绘制

1 个答案: