我用matplotlib,mplot3d和gdal做了一个表面图。这是代码:
import gdal
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
import matplotlib.pyplot as plt
import numpy as np
# maido is the name of a mountain
# tipe is the name of a french school project
# 1) opening maido geotiff as an array
maido = gdal.Open('dem_maido_tipe.tif')
dem_maido = maido.ReadAsArray()
# 2) transformation of coordinates
columns = maido.RasterXSize
rows = maido.RasterYSize
gt = maido.GetGeoTransform()
x = (columns * gt[1]) + gt[0]
y = (rows * gt[5]) + gt[3]
X = np.arange(gt[0], x, gt[1])
Y = np.arange(gt[3], y, gt[5])
# 3) creation of a simple grid without interpolation
X, Y = np.meshgrid(X, Y)
# 4) deleting the "no data" values
# delete the last column
dem_maido = np.delete(dem_maido, len(dem_maido)-1, axis = 0)
X = np.delete(X, len(X)-1, axis = 0)
Y = np.delete(Y, len(Y)-1, axis = 0)
# delete the last row
dem_maido = np.delete(dem_maido, len(dem_maido[0])-1, axis = 1)
X = np.delete(X, len(X[0])-1, axis = 1)
Y = np.delete(Y, len(Y[0])-1, axis = 1)
# 5) plot the raster
fig, axes = plt.subplots(subplot_kw={'projection': '3d'})
surf = axes.plot_surface(X, Y, dem_maido, rstride=1, cstride=1, cmap=cm.gist_earth,linewidth=0, antialiased=False)
plt.colorbar(surf) # adding the colobar on the right
plt.show()
以下是我使用的GeoTiff文件(它在我自己的Google帐户上传,请不要担心):https://drive.google.com/file/d/0B7P95aWmH4DUQk9SbzhNUVNINGs/view
以下是我得到的内容:https://i.stack.imgur.com/hvFed.png
我对结果感到满意,但是当我尝试移动它时,需要很长时间。
我可以做些什么来加快速度?