用Python的matplotlib，如何繪製用pcolor創建的單元格中心的一個點？

Question

我使用matplotlib.pyplot.pcolor繪製二維網格類似如下：

PL.pcolor(array, cmap = PL.cm.YlOrRd, vmin = 0, vmax = 3) 

哪裏array只是一個簡單的二維數組。這部分工作正常。但接下來我嘗試：

PL.hold(True) 
PL.scatter(x, y, 'blue') 
PL.hold(False) 

凡x和y是我想要一個給定的點被繪製的座標。然而，點不是繪製在與pcolor一起繪製的網格單元的中心，而是到達單元格的角落（無論選擇哪個單元格）。

Answer 1

您可以使用meshgrid來生成點的網格。如果您的採樣x和y是均勻的，那麼只需在每個方向上加上分隔2。即，如果分離是1，再加入0.5和減1分：

import numpy as np 
import matplotlib.pyplot as plt 
r = np.arange(10) 
p = np.arange(10) 
R,P = np.meshgrid(r,p) 
data = np.random.random((10,10)) 
plt.pcolor(R,P,data) 
plt.scatter(R[:-1,:-1]+0.5,P[:-1,:-1]+0.5, color = 'blue') 

這是如果你使用tripcolor： 你可以計算你點的質心：triang.triangles給你三點每個三角形的索引，然後使用這三個點計算質心。即centroidX = (x1+x2+x3)/3.; centroidY = (y1+y2+y3)/3.;

import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.tri as tri 
x = np.random.random(10) 
y = np.random.random(10) 
z = np.random.random(10) 
triang = tri.Triangulation(x, y) 
plt.tripcolor(triang,z) 
centroidX = [x[i].sum()/3. for i in triang.triangles] 
centroidY = [y[i].sum()/3. for i in triang.triangles] 
plt.scatter(centroidX,centroidY, color = 'blue')