Python matplotlib PathCollection.legend_elements用法及代碼示例

本文簡要介紹 python 語言中matplotlib.collections.PathCollection.legend_elements的用法。

用法

legend_elements(prop='colors', num='auto', fmt=None, func=<function PathCollection.<lambda>>, **kwargs)

為 PathCollection 創建圖例句柄和標簽。

每個圖例句柄都是一個 Line2D ，代表所繪製的路徑，每個標簽都是一個字符串，每個路徑代表什麽。

這對於獲取 scatter 圖的圖例很有用；例如：

scatter = plt.scatter([1, 2, 3],  [4, 5, 6],  c=[7, 2, 3])
plt.legend(*scatter.legend_elements())

創建三個圖例元素，每種顏色一個，其數值作為標簽傳遞給c。

另請參閱自動圖例創建示例。

參數：

prop {"colors", "sizes"}，默認："colors"

如果"colors"，圖例句柄將顯示集合的不同顏色。如果"sizes"，圖例將顯示不同的大小。要同時設置兩者，請使用kwargs 直接編輯 Line2D 屬性。

num int、None、"auto"(默認)、類似數組或 Locator

要創建的目標元素數。如果沒有，則使用可映射數組的所有唯一元素。如果是整數，則目標是使用規範範圍內的num 元素。如果 "auto" ，嘗試確定哪個選項更適合數據的性質。由於使用 Locator 來查找有用的位置，創建的元素數量可能與num 略有不同。如果是列表或數組，請使用這些元素作為圖例。最後，可以提供 Locator 。

fmt str、 Formatter 或無(默認)

用於標簽的格式或格式化程序。如果字符串必須是 StrMethodFormatter 的有效輸入。如果 None (默認)，使用 ScalarFormatter 。

func 函數，默認：lambda x: x

計算標簽的函數。通常， scatter 的大小(或顏色)參數將由用戶使用函數s = f(x) 進行預處理以使標記可見；例如size = np.log10(x) 。在此處提供此函數的逆函數允許反轉預處理，以便圖例標簽具有正確的值；例如func = lambda x: 10**x 。

**kwargs

允許的關鍵字參數是 color 和 size 。例如：如果使用prop="sizes"，設置標記的顏色可能很有用；如果使用prop="colors"，則類似於設置標記的大小。任何其他參數都將傳遞到 Line2D 實例。這可能對例如有用為圖例句柄指定不同的 markeredgecolor 或 alpha。

返回：

handles Line2D 列表

圖例中每個元素的可視化表示。

labels str 列表

圖例元素的字符串標簽。