如何在 Matplotlib 中为轮廓图创建图例?
matplotlibserver side programmingprogramming更新于 2025/4/30 18:22:17
要在 matplotlib 中为轮廓图创建图例，我们可以采取以下步骤−
- 创建 x、y 和 z 数据点来绘制轮廓函数。
- 要创建 3D 填充轮廓图，我们可以使用 contourf() 方法，使用 x、y、z 和不同级别。
- 使用返回的轮廓签名集合创建矩形列表并设置面颜色
- 现在，使用代理(步骤 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()
输出
