python - 如何绘制这个有很多函数的图的图例？

Question

我有一个矩阵 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

非常感谢你的帮助!

Answer 1

我会说这是方法:

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

plt.legend()

这使得这个情节:



但我在问自己，这是否是你要找的。