Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib(Python)로 그래프 선에 레이블 지정하는 방법

Matplotlib에서 그래프의 선에 레이블을 지정하려면 plot() 메서드의 인자로 label을 전달하면 됩니다. 이렇게 지정한 레이블은 범례(legend)를 통해 그래프에 표시할 수 있습니다.

단계별 진행 방법

  • 그림(figure) 크기를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
  • plot() 메서드에 label="line1"을 지정하여 첫 번째 선을 그립니다.
  • plot() 메서드에 label="line2"을 지정하여 두 번째 선을 그립니다.
  • 그림에 범례를 배치하기 위해 legend() 메서드를 호출합니다.
  • 그래프를 화면에 표시하기 위해 show() 메서드를 사용합니다.

예제 코드

from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
line1, = plt.plot([1, 2, 3], label="line1")
line2, = plt.plot([3, 2, 1], label="line2")
leg = plt.legend(loc='upper center')
plt.show()

실행 결과

위 코드를 실행하면 두 개의 선이 각각 "line1"과 "line2"라는 레이블과 함께 표시되며, 그래프 상단 중앙에 범례가 나타납니다.

Matplotlib(Python)로 그래프 선에 레이블 지정하는 방법

추가 팁

legend() 메서드의 loc 매개변수를 활용하면 범례 위치를 자유롭게 조절할 수 있습니다. 예를 들어 'upper right', 'lower left' 등의 값을 사용할 수 있으며, loc='best'로 설정하면 Matplotlib가 데이터와 겹치지 않는 최적의 위치를 자동으로 선택해 줍니다. 또한 각 선에 다양한 색상이나 선 스타일을 지정하면 범례와 함께 시각적으로 더욱 구분하기 쉬운 그래프를 만들 수 있습니다.