如何在 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+ 瀏覽量

開啟您的職業生涯

完成課程以獲得認證

開始
廣告
© . All rights reserved.