[파이썬 matplotlib] 그래프에 범례(legend) 추가하기

[파이썬 matplotlib] 

그래프에 범례(legend) 추가하기

그래프에 범례를 추가할 때는 legend 함수를 사용합니다. plot 함수 안에 입력한 label 이 범례로 사용됩니다. 

import numpy as np

import matplotlib.pyplot as plt

X = np.linspace(0, 100, 100)

Y1 = X**2

Y2 = X**2+5000

plt.plot(X,Y1, label='Y1')

plt.plot(X,Y2, label='Y2')

plt.legend()

plt.show()

아래와 같이 legend 함수 안에 label 을 입력할 수도 있습니다. 

import numpy as np

import matplotlib.pyplot as plt

X = np.linspace(0, 100, 100)

Y1 = X**2

Y2 = X**2+5000

plt.plot(X,Y1)

plt.plot(X,Y2)

plt.legend(('Y1','Y2'))

plt.show()