如何通过imshow在python中绘制地理标志数据

Question

我有一个带有高程数据init的地理标志光栅数据集。光栅图像中没有数据由-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()