2015-09-08 192 views
4

我想繪製變形的矩形網格,這意味着節點的座標取決於節點的索引。目的是通過函數來​​可視化單位平方的變形。使用Python繪製變形2D網格

我該如何在python中做到這一點?

回答

6

這是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() 

enter image description here

在一個側面說明,pcolormesh默認使用無抗鋸齒性能方面的原因。如果添加antiailiased=Truepcolormesh電話,你會得到一個更好看的結果:

enter image description here