Как построить легенду этого графика, у которого много функций?

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()

, который делает этот график:

Но я спрашиваю я, если это то, что вы ищете.