用Python绘制变形二维网格

Question

我想绘制一个变形的矩形网格，这意味着节点的坐标取决于节点的索引。目的是用一个函数来可视化单位面积的变形。在

我如何在python中做到这一点呢？在

Answer 1

这是pcolormesh（或pcolor）的用途。（还可以查看三角形网格的triplot等。）

import matplotlib.pyplot as plt

y, x = np.mgrid[:10, :10]
z = np.random.random(x.shape)

xdef, ydef = x**2, y**2 + x

fig, axes = plt.subplots(ncols=2)
axes[0].pcolormesh(x, y, z, cmap='gist_earth')
axes[1].pcolormesh(xdef, ydef, z, cmap='gist_earth')

axes[0].set(title='Original', xticks=[], yticks=[])
axes[1].set(title='Deformed', xticks=[], yticks=[])

plt.show()

另外，出于性能考虑，pcolormesh默认不使用抗锯齿。如果您将antiailiased=True添加到pcolormesh调用中，您将得到一个更好看的结果：