我在使用Python的Basemap模块绘制带有经纬度数据的RGB图像时遇到了一个问题。现在,我可以绘制我想要的图,但问题是它有多慢,因为它能够绘制单通道数据比RGB数据快得多,而且通常,单独绘制RGB图像也很快。因为我有lat/lon数据,所以事情变得复杂起来。我已经找到了这个问题的解决方案:
这就是我现在的处境。基本上可以归结为以下问题。在basemap中使用pcolormesh方法时,要绘制RGB数据,必须定义一个colorTuple参数,该参数将逐点映射RGB数据。由于数组大小大约为2000x1000,所以这需要一段时间。下面是我所说内容的一个片段(下面是完整的工作代码):if one_channel:
m.pcolormesh(lons, lats, img[:,:,0], latlon=True)
else:
# This is the part that is slow, but I don't know how to
# accurately plot the data otherwise.
mesh_rgb = img[:, :-1, :]
colorTuple = mesh_rgb.reshape((mesh_rgb.shape[0] * mesh_rgb.shape[1]), 3)
# What you put in for the image doesn't matter because of the color mapping
m.pcolormesh(lons, lats, img[:,:,0], latlon=True,color=colorTuple)
当只绘制一个通道时,它可以在大约10秒左右的时间内绘制出地图。绘制RGB数据时,可能需要3-4分钟。特别是当数据以3倍的速度绘制时,你会感觉到数据是矩形的。在
所以,我的问题是:有没有任何方法可以使这个计算更快,或者使用其他绘图模块(例如Bokeh)或通过任何方式改变颜色映射?我尝试过使用imshow和精心选择的地图边界,但由于它只是将图像拉伸到地图的全部范围,这对于数据的精确映射来说还不够好。在
下面是我的代码的精简版本,它将用于使用正确模块的示例:
^{pr2}$