文章目录
[ Matplotlib version: 3.2.1 ]
六、密度图与等高线图
在二维图上用等高线图或者彩色图来表示三维数据是个不错的方法。
Matplotlib提供三个函数来解决这个问题:
- 用
plt.contour
画等高线图 - 用
plt.contourf
画带有填充色的等高线图(filled contour plot)的色彩 - 用
plt.imshow
显示图形
%matplotlib inline
import matplotlib.pyplot as plt
plt.style.use('seaborn-white')
import numpy as np
三维函数的可视化
1. 用等高线图可视化三维数据
# 首先用函数z=f(x,y)演示一个等高线图
def f(x, y):
return np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)
等高线图可以用plt.contour
函数来创建。它需要三个参数:x轴、y轴、z轴三个坐标轴的网格数据。x轴与y轴表示图形中的位置,而z轴将通过等高线的等级来表示。
用np.meshgrid
函数准备数据是最简单的方法,它可以从一维数组构建二维网格数据<