【python】等高线图

最新推荐文章于 2025-04-04 20:24:26 发布

原创最新推荐文章于 2025-04-04 20:24:26 发布 · 652 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

python 专栏收录该内容

16 篇文章

订阅专栏

本文展示了如何使用Python的matplotlib库创建不同类型的contour图，包括常规线图、等值线图和填充图，以及如何结合numpy生成数据矩阵进行可视化分析。

部署运行你感兴趣的模型镜像

先看一下效果：
在这里插入图片描述

在这里插入图片描述

代码：

import numpy as np
import matplotlib.pyplot as plt


def contour(X, Y, Z):
    plt.style.use('seaborn-white')
    plt.figure(1)
    plt.contour(X, Y, Z, colors='black')
    plt.figure(2)
    plt.contour(X, Y, Z, 20, cmap='RdGy')
    plt.figure(3)
    plt.contourf(X, Y, Z, 20, cmap='RdGy')
    plt.colorbar()
    plt.figure(4)
    contours = plt.contour(X, Y, Z, 3, colors='black')
    plt.clabel(contours, inline=True, fontsize=8)
    plt.imshow(Z, extent=[0, 5, 0, 5], origin='lower',
               cmap='RdGy', alpha=0.5)
    plt.colorbar()
    plt.show()


if __name__ == "__main__":

    x = np.linspace(0, 5, 50)
    y = np.linspace(0, 5, 40)
    X, Y = np.meshgrid(x, y)
    Z = np.sin(X) ** 10 + np.cos(10 + Y * X) * np.cos(X)
    # z的大小应该是x行y列
    contour(X, Y, Z)