python实现dem输出三维模型_python DEM数据可视化

我们在进行地形拟合，或者进行地形分析的时候，尝尝需要将DEM数据可视化，来于其他数据进行比较。下面是pyrhon DEM数据可视化代码

if __name__ == '__main__':

filePath = u"D:/test/fang" # 输入你的dem数据

dataset = gdal.Open(filePath)

adfGeoTransform = dataset.GetGeoTransform()

band = dataset.GetRasterBand(1) # 用gdal去读写你的数据，当然dem只有一个波段

nrows = dataset.RasterXSize

ncols = dataset.RasterYSize # 这两个行就是读取数据的行列数

Xmin = adfGeoTransform[0] # 你的数据的平面四至

Ymin = adfGeoTransform[3]

Xmax = adfGeoTransform[0] + nrows * adfGeoTransform[1] + ncols * adfGeoTransform[2]

Ymax = adfGeoTransform[3] + nrows * adfGeoTransform[4] + ncols * adfGeoTransform[5]

x = np.linspace(Xmin, Xmax, ncols)

y = np.linspace(Ymin, Ymax, nrows)

X, Y = np.meshgrid(x, y)

Z = band.ReadAsArray(0, 0, nrows, ncols