使用 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()
输出
广告