Matplotlib은 파이썬으로 데이터 시각화를 할 수 있는 강력한 라이브러리입니다. 데이터를 그래프로 표현할 때, 그래프의 요소들이 무엇을 나타내는지 알려주는 범례가 중요합니다. 이 글에서는 Matplotlib을 사용하여 그래프에 범례를 추가하는 방법에 대해 알아보겠습니다.

1. Matplotlib에서 기본적인 그래프 그리기

우선 Matplotlib을 사용하여 기본적인 그래프를 그려봅시다. 아래는 간단한 선 그래프를 그리는 예제 코드입니다.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.plot(x, y, label='Prime Numbers')
plt.show()

위 코드를 실행하면 x와 y값을 이용하여 선 그래프가 그려집니다.

2. 그래프에 범례 추가하기

이제 위의 그래프에 범례를 추가해보겠습니다. plt.plot() 함수 호출 시 label 파라미터에 각 데이터의 의미를 나타내는 이름을 지정합니다. 그런 다음 plt.legend() 함수를 호출하여 범례를 표시합니다.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.plot(x, y, label='Prime Numbers')
plt.legend()
plt.show()

위 코드를 실행하면 그래프에 “Prime Numbers”라는 범례가 표시됩니다.

3. 범례 위치 지정하기

때로는 그래프 상에서 범례의 위치를 지정해야할 경우가 있습니다. plt.legend() 함수의 loc 파라미터를 사용하여 범례의 위치를 지정할 수 있습니다. 예를 들어, 아래와 같이 loc='upper left'로 설정하면 범례가 그래프의 왼쪽 상단에 표시됩니다.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.plot(x, y, label='Prime Numbers')
plt.legend(loc='upper left')
plt.show()

위의 예제 코드를 실행하여, 그래프 상에서 범례의 위치를 확인해보세요.

Matplotlib을 사용하면 그래프에 범례를 쉽게 추가할 수 있습니다. 코드를 실행해보고 범례의 위치를 조절해보면서 Matplotlib 라이브러리를 활용하여 데이터 시각화를 더욱 효과적으로 할 수 있습니다.