python - 如何遍历 MeshGrid？

Question

我有一个函数可以返回点 (x, y) 的密度估计。我想遍历给定二维网格的所有 (x, y) 点，并让密度​​函数计算每个点的估计值，以便我可以有一个密度值矩阵，然后我可以绘制它。
假设函数被调用 density(x, y) , 取任意点 (x, y) 并返回该点 (x, y) 的密度估计值 (z)。我希望能够将该函数应用于二维网格中的每个点并存储我可以使用的密度估计，例如 plt.pcolormesh()查看密度。
我怎样才能做到这一点？

Answer 1

我想你想要这样的东西。
首先，定义一个密度函数。为简单起见，我采用函数 |x| + |y| .

def density(x, y):
    return np.abs(x) + np.abs(y)

现在让我们定义沿 x 的点和 y维度并填充数组。在以下示例中，x和 y是存储 n_x 的一维数组和 n_y每个点均在 [-1, 1] 中均匀采样.

n_x = 100
n_y = 100
x = np.linspace(-1, 1, n_x)
y = np.linspace(-1, 1, n_y)

根据点对计算网格并计算密度 D在网格中的每个点上。

xx, yy = np.meshgrid(x, y)
D = density(xx, yy)

请注意，您不需要显式迭代meshgrid，您可以使用看似标量density()数组函数 xx和 yy以及。详情meshgrid ，看到这个 page .
接下来只需使用 pcolormesh()显示或保存。

plt.pcolormesh(x, y, D)
plt.title('Density function = |x| + |y|')
plt.savefig('density.png')

输出是: