matplotlib, добавить легенду для каждой строки? [Дубликат]

Добавить ярлыки для каждого аргумента в вашем графическом вызове, соответствующие графику, например label = "series 1"

. Затем просто добавьте Pyplot.legend() в конец вашего скрипта, и в легенде будут отображаться эти метки .

Вы можете получить доступ к экземпляру Axes (ax) с помощью plt.gca(). В этом случае вы можете использовать

plt.gca().legend()

. Вы можете сделать это либо с помощью ключевого слова label= в каждом из ваших вызовов plt.plot(), либо путем присвоения ваших меток в виде кортежа или списка внутри legend, как в этом рабочем примере:

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-0.75,1,100)
y0 = np.exp(2 + 3*x - 7*x**3)
y1 = 7-4*np.sin(4*x)
plt.plot(x,y0,x,y1)
plt.gca().legend(('y0','y1'))
plt.show()

Однако, если вам нужно получить доступ к экземпляру Axes больше, чем однажды, я рекомендую сохранять его к переменной ax с

ax = plt.gca()

, а затем вызывая ax вместо plt.gca().

Вот пример, чтобы помочь вам ...

fig = plt.figure(figsize=(10,5))
ax = fig.add_subplot(111)
ax.set_title('ADR vs Rating (CS:GO)')
ax.scatter(x=data[:,0],y=data[:,1],label='Data')
plt.plot(data[:,0], m*data[:,0] + b,color='red',label='Our Fitting 
Line')
ax.set_xlabel('ADR')
ax.set_ylabel('Rating')
ax.legend(loc='best')
plt.show()