Matplotlib-网格网格

Question

我想为网格的解决方案制作一个曲面图。我的域范围从 x 和 y 方向的 [-1, 1] 。我了解我需要为 x 和 y 的 plot_surface(x,y,z) 提供二维数组。但是，我不明白如何为 z 安排这个。

为了让事情更清楚，假设我在每个方向上采用 3 个点的网格间距。我最终分别在 x 和 y 方向上得到 [-1, 0, 1] 和 [-1, 0, 1] 。这给了我 9 分，我需要评估 z。这是我写的：

x = np.linspace(-1,1, 3)
y = np.linspace(-1,1, 3)
X,Y = np.meshgrid(x, y)
# Remaining code uses plot_surface(X, Y, Z_approx)
#

每个 X 和 Y 都是 2D 数组，每个长度为 3。但是，我的 Z“近似值”的长度为 9。然后如何将其绘制在一起？

这是网格的外观！

*(-1,1) ____ *(0,1) ____ *(1,1)

*(-1,0) ____ *(0,0) ____ *(1,0)

*(-1,-1)____ *(0,-1)____ *(1,-1)

Answer 1

当我们不知道您的 z 矩阵是什么时，提供答案有点困难，所以这是我的两个最佳猜测：

1. z 是一个 1x9（或 9x1）向量，在这种情况下，应使用 np.reshape 命令将其重新整形为 3x3 矩阵。

例如

Z = np.reshape(Z_approx, (3, 3))

或

1. z 是一个 9x9 矩阵，在这种情况下，我将使用这个小改动制作 x 和 y 9x9 矩阵而不是 3x3。

例如

x = np.linspace(-1,1, 9)
y = np.linspace(-1,1, 9)

希望对您有所帮助，或者您可以让我们了解更多有关您的 z 矩阵的信息。