python matplotlib 绘制等高线

python matplotlib 绘制等高线

""" 绘制等高线 """ import numpy as np import matplotlib.pyplot as mp # 准备数据 num = 500 x, y = np.meshgrid(np.linspace(-3, 3, num), np.linspace(-3, 3, num)) print(x[:4], "-->X") print(y[:4], "-->y") z = (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2) * 10 # 创建窗口 mp.figure("Contour", facecolor="lightgray", figsize=(8, 6)) mp.title("Contour", fontsize=20) # 等高线 颜色映射 mp.contourf(x, y, z, 10, cmap="jet") # 绘制等高线 cnt_r = mp.contour(x, y, z, 10, colors="b", linewidths=0.5) mp.clabel(cnt_r, inline_spacing=0.1, fmt="%.2f", fontsize=8) # 等高线标签值 mp.show()