Matplotlib：繪制自定義圖例

Question

我遇到的情況是，我在matplotlib圖上繪制了內容，而散布圖形狀映射到了標簽類型。

例如：

's': "X"
'o': "Y"

但是，我用網絡x作圖。 現在，我想添加一個顯示此映射的圖例，並且可以使用plt方法修改圖。

所以我需要以某種方式plt.legend以下內容：

plt.legend(('s','o'),("X","Y"))

但是從文檔中不清楚如何完成此任務。 有沒有辦法在matplotlib中繪制這樣的自定義圖例？

Answer 1

從您的問題中不清楚映射的含義。 如果您希望將圖例手柄從默認標記修改為自定義變量標記，則可以執行以下操作。 我的解決方案基於此答案，但簡化為一個簡單的案例。 不要忘記贊成原始答案。 我已經承認了。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

fig, ax = plt.subplots()

new_legends = ["X", "Y"]
markers = ['s', 'o']
colors = ['r', 'b']

x = np.arange(5)
plt.plot(x, 1.5*x, marker=markers[0], color=colors[0], label='squares')
plt.plot(x, x, marker=markers[1], color=colors[1], label='circles')

_, labels = ax.get_legend_handles_labels()

def dupe_legend(label, color):
    line = Line2D([0], [0], linestyle='none', mfc='black',
                mec=color, marker=r'$\mathregular{{{}}}$'.format(label))
    return line

duplicates = [dupe_legend(leg, color) for leg, color in zip(new_legends, colors)]
ax.legend(duplicates, labels, numpoints=1, markerscale=2, fontsize=16)
plt.show()

Answer 2

在Axis對象上繪制內容時，可以傳遞label關鍵字參數：

fig, ax = plt.subplots(figsize=(8, 8))

n_points = 10
x1 = np.random.rand(n_points)
y1 = np.random.rand(n_points)
x2 = np.random.rand(n_points)
y2 = np.random.rand(n_points)

ax.scatter(x1, y1, marker='x', label='X')
ax.scatter(x2, y2, marker='o', label='y')
ax.legend()

這將產生如下結果：