python - 使用 Matplotlib a*y + b*x + c 绘制 3D 曲面

Question

我是 Matplotlib 新手，我需要在 3d 图中绘制一个平面。我在方程中得到了 a、b 和 c 的值，类似于 1y + 2x + 3。

theta = np.array([1,2,3])
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(theta[0],theta[1],theta[2])
plt.show()

我知道这不是使用 plot_surface() 函数的正确方法，但我不知道如何使用。

更新 1

我用线框想出了一些办法。

# Plot the plane
X = np.linspace(0,100, 500)
Y = np.linspace(0,100, 500)
Z = np.dot(theta[0],X) + np.dot(theta[1],Y) + theta[2]
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(X,Y,Z)
plt.show()

但它只显示一条线。

Answer 1

试试这个:

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
Z = X + Y * 2 + 3
# Plot the surface.
ax.plot_surface(X, Y, Z, linewidth=0)
plt.show()

您需要首先创建变量的网格，然后计算网格上的函数值。