Python의 Matplotlib 라이브러리에서 plt.legend() 메서드를 사용하면 그래프에 범례(legend)를 간편하게 추가할 수 있습니다. 특히 frameon 매개변수를 True로 설정하면 범례 주위에 테두리가 함께 표시되어 가독성이 더욱 좋아집니다.
구현 단계
plt.xlabel()메서드를 사용하여 X축 라벨을 설정합니다.plt.ylabel()메서드를 사용하여 Y축 라벨을 설정합니다.plot()메서드를 사용하여 선 그래프를 그립니다.loc매개변수로 범례의 위치를 지정하고,frameon플래그를True로 설정하여 범례 테두리를 활성화합니다.- "blue"와 "orange" 항목으로 범례를 구성합니다.
plt.show()메서드를 사용하여 최종 그림을 화면에 출력합니다.
예제 코드
import matplotlib.pyplot as plt
plt.ylabel("Y-axis ")
plt.xlabel("X-axis ")
plt.plot([9, 5], [2, 5], [4, 7, 8])
location = 0 # 최적의 위치 자동 지정
legend_drawn_flag = True
plt.legend(["blue", "orange"], loc=0, frameon=legend_drawn_flag)
plt.show()
실행 결과
위 코드에서 loc=0은 Matplotlib가 겹치지 않는 최적의 위치를 자동으로 선택하도록 하는 옵션입니다. 만약 범례를 원하는 위치에 고정하고 싶다면 loc='upper right', loc='lower left' 등 다양한 문자열 값을 사용할 수도 있습니다.