如何绘制这个有很多功能的图的图例？

Posted 2023-02-16

【中文标题】如何绘制这个有很多功能的图的图例？

【英文标题】：How to plot the legend of this graph of which there are many functions?

【发布时间】：2020-10-09 01:13:01

【问题描述】：

我有一个矩阵 X 仅包含 1 行用作输入。我有一个矩阵Y，其中每一行都用作函数的输出。如果Y 有 5 行，那么我们就有 5 个不同的函数。我使用以下代码在同一张图上绘制所有函数：

import numpy as np
import matplotlib
import matplotlib.pyplot as plt

Y = np.random.randint(1, 9, size = (5, 7))
X = np.arange(- 3, 4)

fig, ax = plt.subplots()
for i in range(5):
    ax.plot(X, Y[i])

及其结果

我想问一下如何添加这个图表的图例。我想要的格式是

n = 0: color 0
n = 1: color 1
n = 2: color 2
n = 3: color 3
n = 4: color 4
n = 5: color 5

非常感谢您的帮助！

【参考方案1】：

我会说是这样：

for i in range(5):
    ax.plot(X, Y[i], label = i)

plt.legend()

这使得这个情节：

但我在问自己这是否是您正在寻找的东西。

【讨论】：

非常感谢！这正是我正在寻找的:))