【问题标题】:Python quiver and pcolormesh not lining up exactly rightPython quiver 和 pcolormesh 没有完全正确排列
【发布时间】:2016-11-30 22:22:56
【问题描述】:

我试图用风速的 pcolormesh 在地图上覆盖风场的颤抖图。

from mpl_toolkits.basemap import Basemap
from pylab import * 

lonMin = 115.5
lonMax = 124.5
latMin = 10
latMax = 20

res = 0.25
lonGrid = arange(lonMin, lonMax, res)
latGrid = arange(latMin, latMax, res)
lonGrid,latGrid = meshgrid(lonGrid,latGrid)

u = random(lonGrid.shape)
v = random(lonGrid.shape)
m = Basemap(llcrnrlon=lonMin,llcrnrlat=latMin,urcrnrlon=lonMax,urcrnrlat=latMax, resolution = 'i')

m.pcolormesh(lonGrid, latGrid, sqrt(u**2+v**2))
m.quiver(lonGrid,latGrid,u,v, latlon = 'true')
m.drawcoastlines()
m.fillcontinents()

这给了我类似下面的情节

我注意到两件事:

  1. 颤动的箭头从 pcolormesh 生成的像素的左下角发出,而不是从中心发出
  2. 绘图的上边界和最右边界没有着色

这里到底发生了什么,我该如何解决?

【问题讨论】:

  • pcolor(mesh) 模仿了同名的 MATLAB 函数,通常不是数据可视化的最佳选择。我个人会使用像contourf 这样的东西,有很多级别。甚至imshow.

标签: python matplotlib plot matplotlib-basemap


【解决方案1】:

您的代码存在一些问题。

首先,避免使用from pylab import *,这会严重污染您的命名空间。

其次,顶部和右侧的缺失数据:这是由于 pcolormesh 的行为,它模仿了同名的 MATLAB 函数。引用 the functionally similar pcolor 的文档对此进行了解释:

pcolor(X, Y, C, **kwargs)

[...]

理想情况下,X 和 Y 的尺寸应该比 C 大一;如果维度相同,则 C 的最后一行和最后一列将被忽略。

因此,您可以通过使用纬度/经度的辅助数组来消除空白边框。或者,我建议使用imshow,其底图版本会自动调整比例,以便绘制的图像跨越可见地图。将您的 pcolormesh 呼叫切换到

m.imshow(sqrt(u**2+v**2),interpolation='none')

你得到

现在,最后一个问题是您如何尝试可视化数据。您的数据是什么?在上图中,数据点对应于每个“像素”的右下角,即您的 (lat,lon) 点所在的位置。所以现在的可视化是这样的:每个箭头从它对应的点开始,每个像素对应它左下角的数据。

您想要做的是以某种方式将这些箭头移动到像素的中心。如果你想要精确,你实际上需要移动像素,因为根据定义,箭袋图应该在哪里。另一种选择是保持您的地图原样,并移动箭袋图(此版本背后的基本原理是您将数据离散化,并且在像素尺度上,放置箭头/像素的位置无关紧要)。

因为在我看来,如果你的箭袋箭头停留在原位会更准确,我建议将整个底图移动半个(lat,lon) 单位,以便像素以实际数据点为中心。您可以通过将pivot='middle' 选项传递给quiver 使其最漂亮:在这种情况下,您的箭头将集中在数据点(位于每个像素的中间)而不是源自所述点:

from mpl_toolkits.basemap import Basemap
#from pylab import * 
from pylab import arange,meshgrid,random,sqrt

lonMin = 115.5
lonMax = 124.5
latMin = 10
latMax = 20

res = 0.25
lonGrid = arange(lonMin, lonMax, res)
latGrid = arange(latMin, latMax, res)
lonGrid,latGrid = meshgrid(lonGrid,latGrid)

u = random(lonGrid.shape)
v = random(lonGrid.shape)
m = Basemap(llcrnrlon=lonMin-res/2,llcrnrlat=latMin-res/2,
            urcrnrlon=lonMax-res/2,urcrnrlat=latMax-res/2,
            resolution='i') # shifted!
# data corresponds to (latGrid,lonGrid)
# basemap plot is shifted with (-res/2,-res/2)
# imshow will automatically use the visible map anyway

m.imshow(sqrt(u**2+v**2), interpolation='none')
m.quiver(lonGrid,latGrid,u,v, latlon='true', pivot='middle')
m.drawcoastlines()
m.fillcontinents()

结果图看起来相当不错,现在也很明显颜色与箭头的大小有关:

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2019-02-27
    相关资源
    最近更新 更多