[파이썬 matplotlib] 분할된 그래프 범례 넣기 (legend)

화면을 분할하여 여러개의 그래프를 그린 경우 범례를 넣는 방법을 예시를 통해 알아봅시다. 

plot 메소드로 그래프를 생성할 때, label 옵션을 지정해주어야 합니다. label 에 입력된 이름이 범례에 출력됩니다. 

import matplotlib.pyplot as plt
import pandas as pd

md=pd.DataFrame( {'X':[1,2,3],'Y1':[2,1,5], 'Y2':[2,4,10] } )

#figure,ax 만들기
fig,ax=plt.subplots(1,2)

#그래프 그리기
ax[0].plot(md['X'],md['Y1'],color='r',label="ax01")
ax[0].plot(md['X'],md['Y2'],color='b',label="ax02")
ax[1].plot(md['X'],md['Y1'],color='c',linestyle='dashed',label='ax11')
ax[1].plot(md['X'],md['Y2'],color='y',linestyle='dashed',label='ax12')

#범례 추가하기
ax[0].legend(loc='upper left')
ax[1].legend(loc='upper left')

plt.show()