[파이썬] 범례(레전드) 추가와 위치 설정

범례(레전드)는 그래프에 대한 설명을 제공하는데 사용되는 텍스트 레이블입니다. 파이썬의 맷플롯립(Matplotlib) 라이브러리를 사용하여 그래프에 범례를 추가하고 원하는 위치에 배치하는 방법을 알아보겠습니다.

1. 범례 추가하기

맷플롯립에서 범례를 추가하려면 legend() 함수를 사용합니다. 다음은 간단한 선 그래프에 범례를 추가하는 예제 코드입니다.

import matplotlib.pyplot as plt

# 데이터 생성
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 8, 27, 64, 125]

# 그래프 그리기
plt.plot(x, y1, label="y = x^2")
plt.plot(x, y2, label="y = x^3")

# 범례 추가
plt.legend()

# 그래프 표시
plt.show()

위 코드에서 plot() 함수로 두 개의 데이터 세트를 그래프로 그린 후 legend() 함수를 호출하여 범례를 추가합니다. 범례 항목은 label 매개변수를 사용하여 지정합니다.

2. 범례 위치 설정하기

범례의 위치는 legend() 함수의 loc 매개변수를 통해 설정할 수 있습니다. loc 매개변수에는 다양한 위치 옵션이 있습니다. 가장 일반적인 위치 옵션은 다음과 같습니다.

다음은 범례를 오른쪽 상단에 위치시키는 예제 코드입니다.

import matplotlib.pyplot as plt

# 데이터 생성
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 8, 27, 64, 125]

# 그래프 그리기
plt.plot(x, y1, label="y = x^2")
plt.plot(x, y2, label="y = x^3")

# 범례 추가 및 위치 설정
plt.legend(loc="upper right")

# 그래프 표시
plt.show()

위 코드에서 legend() 함수의 loc 매개변수에 'upper right'를 지정하여 범례를 오른쪽 상단에 위치시킬 수 있습니다.

이와 같이 맷플롯립을 사용하여 그래프에 범례를 추가하고 위치를 설정할 수 있습니다. 그래프에 대한 자세한 설정을 하려면 맷플롯립의 다른 기능들을 활용해 보세요. 세부적인 설정은 맷플롯립 공식 문서를 참고해 주세요.