用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')