matplotlib的基本用法(九)——绘制等高线图

文章作者：Tyan
博客：noahsnail.com  |  CSDN  |  简书

本文主要使用matplotlib进行等高线图的绘制。

• Demo

import matplotlib.pyplot as pltimport numpy as np# 定义等高线高度函数def f(x, y):    return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(- x ** 2 - y ** 2)# 数据数目n = 256# 定义x, yx = np.linspace(-3, 3, n)y = np.linspace(-3, 3, n)# 生成网格数据X, Y = np.meshgrid(x, y)# 填充等高线的颜色, 8是等高线分为几部分plt.contourf(X, Y, f(X, Y), 8, alpha = 0.75, cmap = plt.cm.hot)# 绘制等高线C = plt.contour(X, Y, f(X, Y), 8, colors = 'black', linewidth = 0.5)# 绘制等高线数据plt.clabel(C, inline = True, fontsize = 10)# 去除坐标轴plt.xticks(())plt.yticks(())plt.show()

• 结果