如何在 Matplotlib 中为等值线图创建图例？

要在 matplotlib 中为等值线图创建图例，我们可以执行以下步骤：

  • 创建 x、y 和 z 数据点以绘制等值线函数。
  • 要创建填充的 3D 等值线图，我们可以使用带有 x、y、z 和不同级别的contourf() 方法。
  • 使用返回的等值线签名集合制作一个矩形列表并设置填充颜色。
  • 现在，使用代理（第 3 步）和不同的标签将图例放在图中。
  • 要显示图形，请使用show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x, y = np.meshgrid(np.arange(10), np.arange(10))
z = np.tan(x ** 2 + y ** 2)
cs = plt.contourf(x, y, z, levels=[2, 3, 4, 6])
proxy = [plt.Rectangle((1, 1), 2, 2, fc=pc.get_facecolor()[0]) for pc in
cs.collections]
plt.legend(proxy, ["C1", "C2", "C3"])
plt.show()

结果

更新时间：06-05-2021

2K+ 浏览量

开启您的职业生涯

完成课程以获得认证

开始
广告