基于Rasterio与matplotlib的DEM三维展示

直接上代码了

>>> from mpl_toolkits.mplot3d import Axes3D
>>> import matplotlib.pyplot as plt
>>> import numpy as np
>>> import rasterio as rio
>>> f=rio.open(r'C:\Users\suns\Desktop\Ex1\dem')
>>> nrows=f.height
>>> nrows
327
>>> ncols=f.width
>>> Xmin=f.bounds.left
>>> Xmin
442189.26993154
>>> Xmax=f.bounds.right
>>> Xmax
444619.26993154
>>> Ymin=f.bounds.bottom
>>> Ymin
4161425.3140877
>>> Ymax=f.bounds.top
>>> Ymax
4163060.3140877
>>> x = np.linspace(Xmin,Xmax, ncols)
>>> y = np.linspace(Ymin,Ymax, nrows)
>>> X,Y = np.meshgrid(x, y)
>>> Z=f.read(1)
>>> print(Z)
[[ 1018.3137207 1021.53942871 1025.41796875 ..., 1079.36437988
1077.99060059 1075.89050293]
[ 1019.75012207 1023.7064209 1027.79187012 ..., 1082.44714355
1079.32910156 1077.33557129]
[ 1021.66424561 1025.62988281 1029.59558105 ..., 1085.37890625
1081.47424316 1079.17553711]
...,
[ 995.26403809 994.10968018 992.90063477 ..., 1112.69433594
1113.80249023 1114.9107666 ]
[ 999.4006958 998.52185059 997.52252197 ..., 1114.14465332
1115.0213623 1116.12963867]
[ 1003.53076172 1002.79931641 1001.77233887 ..., 1115.82263184
1116.56359863 1117.34851074]]

>>> region = np.s_[10:400,10:400]
>>> X, Y, Z = X[region], Y[region],Z[region]
>>> fig = plt.figure()
>>> ax = Axes3D(fig)
>>> ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
<mpl_toolkits.mplot3d.art3d.Poly3DCollection object at 0x0000000004F88940>
>>> plt.show()
>>>