使用 Matplotlib 2D 轮廓绘图添加额外的等值线

若要使用 Matplotlib 2D 轮廓绘图添加额外的等值线，我们可以采取以下步骤 -

• 设置图形大小，并调整子图之间和周围的间距。

• 创建函数f(x, y)以从x和y获取z数据点。

• 使用 numpy 创建x和y数据点。

• 使用 Numpy 列出级别。

• 使用contour()方法制作轮廓图。

• 标记轮廓图并设置图表的标题。

• 若要显示图形，请使用show()方法。

示例

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

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

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = f(X, Y)
levels = np.arange(-1.0, 1.5, 0.25)
CS = plt.contour(X, Y, Z, levels=levels)

plt.clabel(CS, inline=1, fontsize=10)

plt.title('levels = {}'.format(levels.tolist()))

plt.show()

输出

Rishikesh Kumar Rishi

更新时间：10-Aug-2021

1K+ 浏览量

启动你的 事业

通过完成课程获得认证

开始