python等高线图中添加辅助线_Matplotlib 中等高线图（contour）的绘制

言

在此前的文章里，我们介绍了 Numpy 中 meshgrid 函数的应用，并提到了等高线图的使用。

今天，我们来介绍一下 matplotlib 中等高线图的绘制。

先上图来看一下等高线图的绘制效果，是不是很炫啊。

下面我们来分步骤介绍 matplotlib 中等高线图(contour)的绘制。

数据初始化import matplotlib

import numpy as np

import matplotlib.cm as cm

import matplotlib.mlab as mlab

import matplotlib.pyplot as plt

%matplotlib inline

# 定义等高线高度函数

def f(x, y):

return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(- x ** 2 - y ** 2)

# 数据数目

n = 256

# 定义x, y

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

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

# 生成网格数据

X, Y = np.meshgrid(x, y)

注：数据初始化过程中用到了 Numpy 的 meshgrid 函数，有兴趣的同学可以了解下这个函数的使