python生成地形图

在计算机图形学中，生成地形图是一项重要的任务。Python语言可以很好地实现此任务，下面将介绍如何使用Python生成地形图。

# 导入相关库
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 定义地形函数
def terrain(x,y):
    return np.sin(np.sqrt(x ** 2 + y ** 2))

# 初始化三维图形
fig = plt.figure()
ax = Axes3D(fig)

# 定义坐标范围和步长
X = np.arange(-5,5,0.1)
Y = np.arange(-5,0.1)
X,Y = np.meshgrid(X,Y)

# 计算地形坐标
Z = terrain(X,Y)

# 绘制地形图
ax.plot_surface(X,Y,Z,cmap='rainbow')
plt.show()

在上述代码中，我们首先定义了一个地形函数terrain(x,y)，它接受两个参数x和y，并返回一个高度值。接着，我们使用numpy库中的arange函数来定义坐标范围和步长，然后使用meshgrid函数生成所有坐标点的网格矩阵。

定义完坐标范围和步长后，我们调用地形函数来计算每一个坐标点的高度，并将这些高度值保存在Z矩阵中。最后，我们使用Axes3D库来初始化一个三维图形并使用plot_surface函数将地形图绘制出来。

通过上述代码，我们可以生成一张简单的地形图。如果需要生成更加复杂的地形图，可以不断改变地形函数来探究更多的可能性。