python怎么画高程三维网格图_matplotlib绘制等高线图

参考自Matplotlib Python 画图教程 (莫烦Python)(12)_演讲•公开课_科技_bilibili_哔哩哔哩 https://www.bilibili.com/video/av16378354/index_10.html#page=12

"""绘制等高线图"""

import numpy as np

import matplotlib.pyplot as plt

def calcu_elevation(x1, y1): # 计算高程

ele = (1-x1/2+x1**5+y1**3)*np.exp(-x1**2-y1**2)

return ele

n = 256

x = np.linspace(-3, 3, n)

y = np.linspace(-3, 3, n)

X, Y = np.meshgrid(x, y) # 将原始数据变为网格数据

# 填充等高线的颜色, 8是等高线分为几部分

plt.contourf(X, Y, calcu_elevation(X, Y), 8, alpha=0.75, cmap=plt.cm.hot) # 8表示要分几部分等高线

# 画等高线

C = plt.contour(X, Y, calcu_elevation(X, Y), 8, colors='black', linewidth=0.5)

plt.clabel(C, inline=True, fontsize=10) # 添加文字标签 inlins表示等高线是穿过数字还是不穿过

plt.xticks([])

plt.yticks([])

plt.show()