Python密度和轮廓图绘制--Matplotlib详解

密度和轮廓图

有些情况下，我们需要在二维图表中使用轮廓或颜色区域来展示三维的数据（可以设想等高线地图或温度分布图）。Matplotlib 提供了三个有用的函数来处理这项任务：

• plt.contour绘制轮廓图
• plt.contourf来绘制填充区域颜色的图表
• plt.imshow来展示图像

本节会介绍几个使用它们的例子。当然我们还是首先从将需要使用的包导入 notebook 和初始化工作开始：

%matplotlib inline
import matplotlib.pyplot as plt
plt.style.use('seaborn-white')
import numpy as np

a.三维可视化函数

我们首先使用一个简单的函数

绘制一个轮廓图来进行说明，我们用来作为数组广播运算的例子：

def f(x, y):
    return np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)

轮廓图可以使用plt.contour函数进行创建。它接收三个参数：

• x参数代表三维网格的平面横轴坐标
• y参数代表三维网格的平面纵轴坐标
• z参数代表三维网格的高度坐标

最容易用来准备这种网格数据的是np.meshgrid函数，可以将两个一维的数