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()