python - 选择部分颜色图绘制地形

数据


.tif数据DEM是海拔值，代表一个行政区划的催眠图。我上传了here
除了分区之外的区域，我不想在地块上显示。


我的目标

使用plt.pcolormesh绘制此部门的催眠图（PS：我发现在阅读.tif时，plt.imshow()比pcolormesh快得多。我不知道为什么）。

在这里，我展示了一个我从互联网上截取的示例。

http://7xrn7f.com1.z0.glb.clouddn.com/16-3-10/56951530.jpg

我的尝试

### Using GDAL to read the .tif data
from osgeo import gdal
### Read the .tif
pathToRaster = r'./dem.tif'
raster = gdal.Open(pathToRaster,  gdal.GA_ReadOnly)
dem = raster.GetRasterBand(1).ReadAsArray()
dem = dem[::-1]

### Mask the outside value
dem_mask = np.ma.masked_less(dem,0)
plt.pcolormesh(dem_mask,cmap =plt.cm.terrain)

结果

http://7xrn7f.com1.z0.glb.clouddn.com/16-3-10/45669007.jpg

问题

在我的研究部门中，该区域不包含上面的blue所绘制的海洋。

但是我想使用plt.cm.terrain作为我的pcolormesh色彩图，因为它适合这种情况。

因此，我想删除颜色图的蓝色部分，并使用表示纯色的green开始地形。

最佳答案

我现在有两个解决方案。

1.使用vmin设置较小的开始。

>print dem_mask.min()
>print dem_mask.max()
output: 20
        2271
## Set vmin value far smaller than 20
plt.pcolormesh(dem_mask,cmap =plt.cm.terrain,vmin = -800)

2.提取引用this question的颜色图的子集

 import matplotlib.colors as colors
 def truncate_colormap(cmap, minval=0.0, maxval=1.0, n=100):
     new_cmap = colors.LinearSegmentedColormap.from_list(
           'trunc({n},{a:.2f},{b:.2f})'.format(n=cmap.name, a=minval, b=maxval),
            cmap(np.linspace(minval, maxval, n)))
     return new_cmap

 cmap = plt.get_cmap('terrain')
 new_cmap = truncate_colormap(cmap, 0.25, 0.9)
 plt.pcolormesh(dem_mask,cmap =new_cmap,)

结果

http://7xrn7f.com1.z0.glb.clouddn.com/16-3-11/96590416.jpg

关于python - 选择部分颜色图绘制地形，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/35920887/