plot - 如何通过imshow在python中绘制geotiff数据

Question

我有一个带有高程数据初始化的 geotiff 栅格数据集。光栅图像中没有数据由 -9999 定义。当我尝试使用以下代码绘制图时：

import gdal
import numpy as np
from mayavi import mlab

ds = gdal.Open('data.tif')
dem = ds.ReadAsArray()
gt = ds.GetGeoTransform()
ds = None

mlab.imshow(dem)
mlab.colorbar()
mlab.show()

问题是当我绘制图时，它也会绘制 nodata 值。我的问题是如何从光栅图像中排除 -9999 值（或选择要绘制的值范围）。

数据链接如下：

https://drive.google.com/file/d/0B2rkXkOkG7ExR1VsVW5HQXBhSDQ/view?usp=sharing

Answer 1

如果您仍然需要一个干净的解决方案来解决这个问题，我相信您正在寻找的是一个掩码数组，numpy.ma例如：

import gdal
import numpy as np
from mayavi import mlab

ds = gdal.Open('data.tif')
dem = ds.ReadAsArray()

msk = dem==-9999 # boolean array with True at elements to be masked
dem = np.ma.array(data=dem, mask=msk, fill_value=np.nan)

gt = ds.GetGeoTransform()
ds = None

mlab.imshow(dem)
mlab.colorbar()
mlab.show()