■ 그래프에 범례 넣기

이번에는 그래프가 의미하는 바를 구별할 수 있도록 범례(legend)를 넣어보겠습니다.

범례는 보통 두 개 이상의 데이터를 표시할 때 사용합니다. 예를 들어, ‘가’라는 책의 월별 대출 횟수 그래프와 ‘나’라는 책의 월별 대출 횟수 그래프처럼 2개의 그래프가 하나의 좌표 평면에 표현되었을 때, ‘가’는 노란색, ‘나’는 초록색으로 표현되었음을 범례로 알릴 수 있습니다.

범례를 넣기 위해 plot() 함수를 사용할 텐데요. plot() 함수에 label이라는 속성의 레이블 값으로 원하는 문자열을 넣어주고, 그래프를 그리기 전에 legend() 함수를 실행시키면 레이블 값이 범례로 나타납니다. 두 가지 범례를 넣으려면 plot() 함수를 두 개 사용하면 되겠지요.

import matplotlib.pyplot as plt
plt.title('legend')
plt.plot([10, 20, 30, 40], label='asc')    # 증가를 의미하는 asc 범례
plt.plot([40, 30, 20, 10], label='desc')   # 감소를 의미하는 desc 범례
plt.legend()
plt.show()

그림 4-5 범례 실행 결과