Python 绘制遥感数字高程影像（DEM）

from matplotlib import cm
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

img = Image.open('高程.tif')
re_img = np.asarray(img)
fig = plt.figure(figsize=(12, 8))
ax = Axes3D(fig)

X = np.arange(0, ncols, 1)
Y = np.arange(0, nrows, 1)
X, Y = np.meshgrid(X, Y)
Z = re_img 
surf = ax.plot_surface(X, Y, Z, rstride=10, cstride=10, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)