运用 Matplotlib 库绘制密度图与等高线

相似文件 换一批

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

以下内容转自 数析学院，原文内容较多，有需要了解更多的同学可以直接查看原文

本节将向大家演示在 Python 中如何运用 Matplotlib 库绘制密度图与等高线。首先，我们需要在 Python 中导入所需的库与模块，并对 notebook 环境与图表风格进行初步的设置：

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

复制代码

为了更方便后续演示，定义如下复杂函数：

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

复制代码

Numpy 中的函数 meshgrid ，我们构造了一个复杂的点阵网格，用于匹配密度图或等高线的一般运用场景：

1. x = np.linspace(0, 5, 50)
   
2. y = np.linspace(0, 5, 40)
   
3. 
   
4. X, Y = np.meshgrid(x, y)
   
5. Z = f(X, Y)

复制代码

函数 contour 是 Matplotlib 中专用于绘制等高线的函数，我们可以直接将坐标的三个维度作为函数的输入，从而绘制等高线图：

1. plt.contour(X, Y, Z, colors='black');

复制代码

上述 colors 参数用于调整图中等高线的颜色，在实际操作中我们也可以利用 cmap 参数来定义指定的 Colormap，另外在下面的例子中，函数的第四个参数 20 表示等高线的密集程度，我们可以通过调整它来控制图中显示的信息量：

1. plt.contour(X, Y, Z, 20, cmap='RdGy');

复制代码

以上内容转自 数析学院，原文内容较多，有需要了解更多的同学可以直接查看原文

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：等高线