用Matplotlib画等高线图

最新推荐文章于 2023-01-30 16:55:53 发布

Cheuk__Ho

最新推荐文章于 2023-01-30 16:55:53 发布

阅读量1k

点赞数 1

分类专栏： python笔记

本文链接：https://blog.csdn.net/xianzhuohao/article/details/104553097

版权

本文介绍了如何使用Matplotlib的contour和contourf函数绘制等高线图，包括创建坐标网格、显示Z值、生成颜色条等步骤。通过实例展示了np.meshgrid在生成坐标网格和理解Z值中的应用。

摘要由CSDN通过智能技术生成

所用到的函数如下：
matplotlib.pyplot.contour(*args, **kwargs)
matplotlib.pyplot.contourf(*args, **kwargs)
两个函数的keyword是一致的，由于两个函数的返回值相同，所以可以结合使用。前者是绘制等高线，后者是填充等高线。
keyword建议查看官方文档。

导入库

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

第一种是通过数组创建坐标网格与Z值

#生成一个x轴是1~3，y轴是7~8的网格坐标
X = [1, 2, 3]
Y = [7, 8]
Z = [[1, 1, 2],
     [2, 2, 2]]

开始画图

plt.figure()
cs = plt.contour(X, Y, Z
                 , levels=(1.2, 1.4,1.6,1.8)  # 需要显示Z值的等高线,也可用数值控制要显示的等高线数量
                 , colors=('r', 'k', 'blue', "yellow")  # 定义线条的颜色,也可用cmap定义
                 , linestyles=(":", "dashed", "-.", "-")
                 # 定义线条的形状，默认是None（正数为实线，负数为虚线），如果定义了线条颜色则都为实线。
                 # 线条形状有'solid&