【python】等高线图

最新推荐文章于 2024-09-12 14:45:11 发布

nencbskk

最新推荐文章于 2024-09-12 14:45:11 发布

阅读量465

点赞数 8

分类专栏： python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/ebatudou/article/details/135683308

版权

python 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

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

在这里插入图片描述

代码：

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)