python - 如何从列表中绘制多个图例

Question

我正在尝试迭代多个图，每个图都有自己的标签。图例值取自列表“iou”。下面的代码只为 iou=0.2 生成 1 个标签。任何帮助表示赞赏。

iou=[0.2, 0.4, 0.6, 0.8]
from matplotlib import pyplot as plt
fig = plt.figure()
for i in range(0,4):
    p=plt.plot(recall[i], precision[i], marker='+')
    plt.legend(p, iou)
plt.title("PR curves")
plt.xlabel("Recall")
plt.ylabel("Precision")

Answer 1

在循环的每次迭代中，您需要使用 plt.plot 的 label 属性，并且仅在循环调用 plt.legend

from matplotlib import pyplot as plt

iou=[0.2, 0.4, 0.6, 0.8]
fig, ax = plt.subplots(1, 1)
for i in range(0,4):
    ax.plot(recall[i], precision[i], marker='+', label=iou[i])

fig.legend()
ax.set_title("PR curves")
ax.set_xlabel("Recall")
ax.set_ylabel("Precision")

一个最小的工作示例如下:

from matplotlib import pyplot as plt
import numpy as np

iou=[0.2, 0.4, 0.6, 0.8]
precision = np.random.randint(0, 10, size=(4, 10))
fig, ax = plt.subplots(1, 1)
for i in range(0,4):
    ax.plot(np.r_[:10], iou[i] * precision[i], marker='+', label=iou[i])

fig.legend()
ax.set_title("PR curves")
ax.set_xlabel("Recall")
ax.set_ylabel("Precision")

给出以下结果: