python - "Heat map"平均值

Question

我正在制作一个“热图”，其中箱中的颜色是每个箱中某些第三个值的平均值。例如，x 和 y 位置，然后是平均海拔。我的问题是散点和框并不总是重叠(左上角)。我只是错过了一些简单的事情吗？有更好的办法吗？

这是一些可以说明我的问题的工作示例代码。使用 Ipython --pylab

from scipy import stats
x=randint(0,50,60 )
y=randint(0,65,60 )
z=randint(0,20,60)
d = stats.binned_statistic_2d(x, y, z)
extent = [d[1][0], d[1][-1], d[2][0], d[2][-1]]
imshow(d[0], cmap='RdBu_r', extent=extent, interpolation='nearest', origin='lower')
scatter(x, y, s=50, c='0.8', edgecolor='0.8')

输出:/image/Yx4iT.png

Answer 1

我观察到 d[0] 中的 x 和 y 方向翻转了。可能是 imshow 的问题。所以只需进行转置(注意 imshow 中的 d[0].T)...

from scipy import stats

x=randint(0,50,60 )
y=randint(0,65,60 )
z=randint(0,20,60)
d = stats.binned_statistic_2d(x, y, z, statistic='mean')
extent = [d[1][0], d[1][-1], d[2][0], d[2][-1]]
imshow(d[0].T, cmap='RdBu_r', extent=extent, interpolation='nearest', origin='lower')
scatter(x,y, s=50, c=z, edgecolor='0.8')

此外，我将 scatter 的第四个参数更改为 c=z，以便您可以看到散点的实际值。